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

Unable to train XGBoost using a DMatrix created from a GPU dataframe #469

Closed
raghavmi opened this issue Dec 7, 2018 · 0 comments
Closed

Unable to train XGBoost using a DMatrix created from a GPU dataframe #469

raghavmi opened this issue Dec 7, 2018 · 0 comments
Assignees
@kkraus14 @mtjrider

Comments

@raghavmi
Copy link
Contributor

raghavmi commented Dec 7, 2018 •
edited
Loading

If you create a DMatrix directly from a GPU dataframe, then training XGBoost with that DMatrix throws an error.

Workaround: Convert the GPU dataframe to a Pandas dataframe first. Create the DMatrix from this Pandas dataframe. This DMatrix will train fine on XGBoost.

You can reproduce this issue on the latest RAPIDS Docker image using the attached Python code .

GPU DF to DMatrix Conversion Error #2.txt
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@kkraus14 kkraus14

@mtjrider mtjrider

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@kkraus14 @mtjrider @raghavmi