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

[FEA] Support categorical features when serving XGBoost models #389

Open
gfalcone opened this issue May 15, 2024 · 2 comments
Open

[FEA] Support categorical features when serving XGBoost models #389

gfalcone opened this issue May 15, 2024 · 2 comments
Labels
enhancement New feature or request

Comments

@gfalcone
Copy link

gfalcone commented May 15, 2024

Hello !

XGBoost recently enabled developers to use categorical features in its models (Nvidia did an article on that : https://developer.nvidia.com/blog/categorical-features-in-xgboost-without-manual-encoding/).

From what I understand, we can load a XGBoost model trained on categorical features within the FIL_BACKEND.

However, the FIL_BACKEND only supports float in inputs, which means that we have to do some kind of ensemble (PYTHON_BACKEND + FIL_BACKEND) to accepts strings (steps described here : https://github.com/triton-inference-server/fil_backend/blob/main/notebooks/categorical-fraud-detection/Fraud_Detection_Example.ipynb)

It would make things easier to accept strings in the FIL_BACKEND. Would it be possible to do that ?

@gfalcone gfalcone changed the title [FEA] [FEA] Support categorical features when serving XGBoost models May 15, 2024
@hcho3
Copy link
Collaborator

hcho3 commented May 28, 2024

It is currently not possible to accept string inputs in the FIL backend. The feature will require substantial amount of developer resource on our end. (The current model storage does not store label encoding, so it needs to be revamped). I will mark this as a feature request.

@hcho3 hcho3 added the enhancement New feature or request label May 28, 2024
@gfalcone
Copy link
Author

Thank you for your reply @hcho3 !

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

No branches or pull requests

2 participants