SklearnTrainer: multi output model and pipelines

Ray Libraries (Data, Train, Tune, Serve)
1

https://docs.ray.io/en/latest/train/api/doc/ray.train.sklearn.SklearnTrainer.html#ray.train.sklearn.SklearnTrainer

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?

2

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