SklearnTrainer: multi output model and pipelines

It seems that it’s not possible to use any model that produces multiple-output?
Since here the label_column is just one str instead of a list.
Also do you support scikit learn pipelines apart from simple BaseEstimator?

Yes, it supports pipelines. But as you pointed out, it does not support MultipleOutput stuff. Looking at SklearnTrainer, this may just be a matter of extending the API of supporting a list of (instead of one) label column. Are you open to contributing to that?

1 Like