hummingbird icon indicating copy to clipboard operation
hummingbird copied to clipboard

Add support for NGBoost (NGBRegressor/NGBClassifier)

Open leonardomurakami opened this issue 3 years ago • 3 comments

Trying to run hummingbird for NGBRegressor and receiving MissingConverter: Unable to find converter for model type <class 'ngboost.api.NGBRegressor'>. It usually means the pipeline being converted contains a transformer or a predictor with no corresponding converter implemented. Please fill an issue at https://github.com/microsoft/hummingbird.

leonardomurakami avatar Oct 01 '21 19:10 leonardomurakami

Thanks! Added this to the requests list.

I'm assuming you mean this NGBoost. Can you please give us a few more details about your model?

ksaur avatar Oct 01 '21 20:10 ksaur

Ay, sorry for taking so long to answer. Giving some more details, it is basically a boosting model which follows the sklearn API, except, on top of being able to do point estimates, it is able to predict a probabilistic distribution through the pred_dist method

Thanks for adding this to the requests list!

leonardomurakami avatar Oct 12 '21 12:10 leonardomurakami