search

yandex / rep

Machine Learning toolbox for Humans
http://yandex.github.io/rep/
Other
687 stars 144 forks source link

Folding Classifier don't correctly work with mask in report.learning_curve #18

Closed tlikhomanenko closed 9 years ago

tlikhomanenko commented 9 years ago

1) Add message about dataset length (equal or not to the training) during staged_predict_proba 2) Add predict_all (True, False) to report learning curve (when mask will be applied, before or after prediction operation)

arogozhnikov commented 9 years ago

1) Add message about dataset length (equal or not to the training) during staged_predict_proba

Resolved, there message now is predicting with random estimator if using folding scheme and train length doesn't coincide with test. https://github.com/yandex/rep/commit/9dd8eee7132ffa6fa77a373b8622824a699329bc

arogozhnikov commented 9 years ago

Resolved in https://github.com/yandex/rep/commit/5b68483e8cea496845f99616b6c83caaf77f77e5

Now there is special parameter predict_only_masked=True