flennerhag / mlens

ML-Ensemble – high performance ensemble learning
http://ml-ensemble.com
MIT License
843 stars 108 forks source link

cant predict_prob #109

Closed yanqiangmiffy closed 5 years ago

yanqiangmiffy commented 5 years ago

It only predict labels?the result of predict_proba and predict is same

flennerhag commented 5 years ago

predict_proba is just for compatibility with Scikit-learn. Since we need to know up-front if we're going to fit the ensemble using probabilities or not, you must specify this in the add method when you construct the ensemble. You do this using the proba parameter. The tutorial provides a more in-depth walk-through.

flennerhag commented 5 years ago

108 might also be of help to you.