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

catboost tune_grid not working with task_type = "GPU" #27

Open
nsandau opened this issue Aug 25, 2020 · 1 comment
Open

catboost tune_grid not working with task_type = "GPU" #27

nsandau opened this issue Aug 25, 2020 · 1 comment

Comments

@nsandau
Copy link

nsandau commented Aug 25, 2020

I'm getting the following error when trying to use tune_grid with GPU enabled for catboost: "Error: rsm on GPU is supported for pairwise modes only".
I have tried setting mtry to 1 or omitting the argument but I still get the error.

@nsandau
Copy link
Author

nsandau commented Aug 25, 2020

It seems the issue is with the conversion of mtry to a proportion:

if(!is.null(rsm)) {

When omitting or setting mtry = 1 the conversion reassigns rsm to rsm/ncol(x).

It should be a relatively easy fix when mtry_prop becomes a part of dials #20 .

A quick workaround until then is to set mtry = ncol(x)+1.

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