Add support for scikit-learn GeneralizedLinearModels (PoissonRegressor, GammaRegressor, TweedieRegressor)

microsoft / hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.

MIT License

3.34k stars 278 forks source link

Closed RomanBredehoft closed 1 year ago

RomanBredehoft commented 1 year ago

Hello ! It would be great if Hummingbird could support scikit-learn GLMs, and more specifically all three PoissonRegressor, GammaRegressor and TweedieRegressor classes !

Is there currently something planned about it ?

interesaaat commented 1 year ago

Thanks @RomanBredehoft for the suggestion. I think these shouldn't be hard to add, I will work on it.

RomanBredehoft commented 1 year ago

Thanks a lot !