Hummingbird compiles trained ML models into tensor computation for faster inference.
MIT License
Published by ksaur over 4 years ago
This release adds several new operators to both scikit-learn and Onnx.
This release would not have been possible without the following contributors: @ahmedkrmn, @KranthiGV, @TuanNguyen27, @zhanjiezhu
Published by interesaaat over 4 years ago
This release adds several new cool features and bug fixes to Hummingbird!
When selecting the backend to use for conversion, we renamed pytorch
into torch
(to match the module name). [#142]
transform
method is added to the PyTorch container to match the transformer
API of Sklearn. [#148 ]lightgbm_converter
in ONNXMLTOOLS) [#142 ]This release would not have been possible without the following contributors: @ahmedkrmn, @jspisak, and @TuanNguyen27.
Published by ksaur over 4 years ago
This release adds several new operators, an updated API, and contains several documentation fixes.
Special thanks to following contributors: @KranthiGV (DecisionTreeRegressor), @mmbhatk (ExtraTreesRegressor), @bfgray3 (GradientBoostingRegressor), and @ahmedkrmn (HistGradientBoostingClassifier)
Published by interesaaat over 4 years ago
This is the first release for Hummingbird! In this release, Hummingbird supports conversion from scikit-learn, LightGBM and XGBoost models to PyTorch. Currently supported models are listed here.