hummingbird
Hummingbird compiles trained ML models into tensor computation for faster inference.
MIT License
Bot releases are visible (Hide)
Published by ksaur over 4 years ago
This release adds several new operators to both scikit-learn and Onnx.
New Features
- float 64 support [#186]
- Better windows installation support [#179]
New Operators - scikit-learn
- IsolationForest [#191]
- LGBMRanker [#173]
- OneHotEncoder [#193]
- Scaler [#171]
- XBGRanker [#189]
New Operators - Onnx
- ArrayFeatureExtractor [#198]
- Linear Classifier/Regressor [#190, #194]
- Normalizer [#188]
- Scaler [#196]
Credits
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!
API Changes
When selecting the backend to use for conversion, we renamed pytorch into torch (to match the module name). [#142]
New Operators
- HistGradientBoostingRegressor [#135 ]
- LinearRegression [#140 ]
- LinearSVC [#140 ]
- LogisticRegression [#140 ]
- LogisticRegressionCV [#140 ]
- Normalizer [#126]
New Features
-
transformmethod is added to the PyTorch container to match thetransformerAPI of Sklearn. [#148 ] - Support for ONNX models as input (at the moment this feature only works in combination with the
lightgbm_converterin ONNXMLTOOLS) [#142 ] - Generation of ONNX models as output (at the moment this feature only works when a ONNX model is passed as input) [#142]
Credits
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.
New Operators
- DecisionTreeRegressor [#102 ]
- ExtraTreesRegressor [#91 ]
- GradientBoostingRegressor [#88 ]
- HistGradientBoostingClassifier [#87]
Credits
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.
