search

minerva-ml / steppy-toolkit

Curated set of transformers that make your work with steppy faster and more effective :telescope:
MIT License
22 stars 9 forks source link

SklearnClassifier predict_proba #9

Closed kstrzala closed 6 years ago

kstrzala commented 6 years ago

In case of 2-categories classification there should be returned array of shape (n,), while currently is (n,2), which is incompatible with some other non-sklearn models.