search

lensacom / sparkit-learn

PySpark + Scikit-learn = Sparkit-learn
Apache License 2.0
1.15k stars 255 forks source link

Fix issue in bayes predict_proba #64

Closed taynaud closed 8 years ago

taynaud commented 8 years ago

predict_proba map to scikit predict_proba, which call self.predict_proba. As sparkit-learn herits from scikit bayes, this call sparkit-learn predict_proba which fail on numpy or scipy array.

taynaud commented 8 years ago

My previous randomforest PR seems to turn the tests a bit flappy, I have increased the threshold

fulibacsi commented 8 years ago

Thanks for your contribution! Really appreciate it! :)