phausamann / sklearn-xarray

Metadata-aware machine learning.
http://bit.do/sklearn-xarray
BSD 3-Clause "New" or "Revised" License
102 stars 12 forks source link

Wrap predict_proba #27

Closed phausamann closed 6 years ago

phausamann commented 6 years ago

Implement an _ImplementsPredictProbaMixin in common.base with methods predict_proba (passthrough to _call_fitted) and _predict_proba (implementing the call of the underlying estimator for DataArray inputs).

phausamann commented 6 years ago

Also: predict_log_proba and decision_function