How to conduct catboost grid search using GPU in R? Ask Question

Question

I just want to have a grid search using the catboost package in R.

fit_control <- trainControl(method = "cv", number = 4, classProbs = TRUE)

grid <- expand.grid(depth = c(7,8,9,10), learning_rate = c(0.1,0.2,0.3,0.4), iterations = c(10,100,1000))

report <- train(df.scale, as.factor(make.names(as.matrix(tier1))), method = catboost.caret, logging_level = 'Verbose', preProc = NULL, tuneGrid = grid, trControl = fit_control)

I know that the above will search across different values for depth, learning rate, and the number of iterations but I can't figure out where to input the option for the task_type = "GPU". Would appreciate any help on how to specify using the GPU for finding the optimal parameters using R.

Answer 1

It seems like in R programming you are using the caret package so that you can perform the training. So I want to tell you that in that case, it looks like the caret package does not pass any additional arguments to the catboost.train function so it may not support the GPU functionality.