decoupling CLI from training. train_multiclass can be reused in the future