Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

XGBRegressor with objective='count:poisson' causes serialization to fail with no error message #658

Open
ParanoidAltoid opened this issue Oct 5, 2023 · 1 comment

Comments

@ParanoidAltoid
Copy link

http://onnx.ai/sklearn-onnx/auto_examples/plot_pipeline_xgboost.html

Download file found at bottom of page in that link, and run with count:poisson:
pipe = Pipeline([("scaler", StandardScaler()), ("xgb", XGBRegressor(n_estimators=3, objective='count:poisson'))])
Comparing results before and after serializing/deserializing:
predict [0.9388053 0.9388053 0.9388053 0.9388053 0.9388053]
predict [1.13 1.13 1.13 1.13 1.13]

Additional info:

Python 3.10, all versions below are latest from pip, except xgboost which doesn't work at all per #651 (comment)_

xgboost==1.7.0
skl2onnx==1.15.0
onnx==1.14.1
onnxconverter-common==1.14.0
onnxmltools==1.11.2
onnxruntime==1.16.0

@xadupre
Copy link
Collaborator

xadupre commented Dec 18, 2023

The latest release version should work with this objective.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants