-
Notifications
You must be signed in to change notification settings - Fork 88
Open
Description
I have installed thundergbm with cuda 11 support. But when I try to run the thundergbm classifier inside a nested loop in order to do a hyperparameter search, I get the following error:
Python Version: 3.7.8
Microsoft Visual Studio 19: v.16.11.8
Cuda: 11.5
cmake: 3.22.1
Below is the code that I run in a Python Jupyter Notebook:
for n in range(100,700,100):
for d in range(3,16,1):
for c in [0.4,0.5,0.6,0.7,0.8]:
clf = TGBMClassifier(depth=d, n_trees = 1, n_parallel_trees=n, bagging=1, column_sampling_rate=c, objective =
"binary:logistic")
clf.fit(X_train, y_train)
What is the reason?
Metadata
Metadata
Assignees
Labels
No labels
