onnxmltools
onnxmltools copied to clipboard
onnx
ONNXMLTools enables conversion of models to ONNX
Maybe related to #338 Since original onnxmltools only supports conversion of `LGBMRegressor` / `LGBMClassifier`, I added some more code to support conversion of `LGBMRanker`. With this change, I found conversion...
Is there a reason why `binary:hinge` isn't supported as `objective` for **XGBoost**? https://github.com/onnx/onnxmltools/blob/fe55a8ac7738dbe6719de49bdb048b0c13e29971/onnxmltools/convert/xgboost/operator_converters/XGBoost.py#L267-L293
Hi, I'm trying to convert both `XGBRFClassifier` & `XGBRFRegressor` models into `ONNX`, but unfortunately there's no support for such models as I can also see from the source code [here](https://github.com/onnx/onnxmltools/blob/main/onnxmltools/convert/xgboost/operator_converters/XGBoost.py#L214)....
Hello machine learning engineers, i have tried to convert an xgboost regressor multi-output model trained on dataset shape=(1000, 210). on the python runtime the model prediction predicts the correct shape...
We are releasing ONNX 1.16.0. A release branch is created (https://github.com/onnx/onnx/tree/rel-1.16.0). The planned release date is March 25, 2024. Release candidates are also available from TestPyPI: `pip install -i https://test.pypi.org/simple/...
- The pkg is dashed while the lib is underscored. - Add it to the dev requirements. Resolve https://github.com/onnx/onnxmltools/issues/673
There's an issue on binarry classifier when passing a booster object. WrappedBooster will set num_class to 1 as trees//ntrees == 1 https://github.com/casassg/onnxmltools/blob/e7a2ef088fb336800d34bffa628abac79f1ecc14/onnxmltools/convert/xgboost/_parse.py#L100 This makes it such that when we try...
