Rearrange the predict and train mode.

Rewrote the predict and train mode args.

Merge request reports

Loading