FastBDT icon indicating copy to clipboard operation
FastBDT copied to clipboard

Refactored Classifier to an sklearn estimator

Open OmriKaduri opened this issue 5 years ago • 0 comments

Hey,

I've wanted to use your classifier for a multiclass problem. The easiest way I've thought about is using sklearn OneVsRestClassifier and refactoring your Classifier to be an sklearn estimator (by using inheritance from BaseEstimator and ClassifierMixin). Also, I've added predict_proba function.

So I've thought it would be nice to open a PR for anyone want to use this classifier for multiclass problems 😃

OmriKaduri avatar Jul 19 '20 21:07 OmriKaduri