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.

Sign up for GitHub

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

Jump to bottom

Correct way of converting HistGradientBoostingRegressor with categorical_features? #1101

Open
ogencoglu opened this issue Jun 4, 2024 · 2 comments
Open

Correct way of converting HistGradientBoostingRegressor with categorical_features? #1101

ogencoglu opened this issue Jun 4, 2024 · 2 comments
Assignees
@xadupre

Comments

@ogencoglu
Copy link

ogencoglu commented Jun 4, 2024 •
edited
Loading

HistGradientBoostingRegressor supports specification of categorical_features: https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.HistGradientBoostingRegressor.html#sklearn.ensemble.HistGradientBoostingRegressor

What would be the correct initial_type in convert_sklearn(model, initial_types) to achieve this behaviour. Float inputs to ONNX are giving wrong predictions of course.
@ogencoglu
Copy link
Author

Pinging @maximilianeber in case you found out a way

@maximilianeber
Copy link

maximilianeber commented Jun 4, 2024 •
edited
Loading

Pinging @maximilianeber in case you found out a way

Replied in the original thread in #1051. We haven't managed to work around this and stuck to OHE for now.

@xadupre xadupre self-assigned this Jun 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@xadupre xadupre

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@ogencoglu @maximilianeber @xadupre