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

Implement support for categorical variables #25

Open
ogrisel opened this issue Nov 1, 2018 · 0 comments
Open

Implement support for categorical variables #25

ogrisel opened this issue Nov 1, 2018 · 0 comments

Comments

@ogrisel
Copy link
Owner

ogrisel commented Nov 1, 2018

pygbm should accept pandas.DataFrame instances as in X in gbm.fit(X, y) (as well as predict, in addition to numpy arrays).

This would make it more natural to accept heterogeneously typed features in the training data.

I am not sure what is the best strategy is to deal with categorical variables. Reviewing the strategy implemented in LightGBM and XGBoost would be helpful.

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

1 participant