update regressor/clasifier configs

EpistasisLab · Sep 3, 2020 · a45046d · a45046d
1 parent ca22319
commit a45046d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 18 deletions.
diff --git a/tpot/config/classifier_cuml.py b/tpot/config/classifier_cuml.py
@@ -37,15 +37,6 @@
         "weights": ["uniform",],
     },
 
-    "cuml.ensemble.RandomForestClassifier": {
-        "n_estimators": [100, 300],
-        "split_algo": [0, 1],
-        "max_depth": range(8, 16),
-        "max_features": np.arange(0.05, 1.01, 0.05),
-        "min_rows_per_node": range(2, 21),
-        "n_bins": [64,]
-    },
-
     "cuml.linear_model.LogisticRegression": {
         "penalty": ["l1", "l2", "elasticnet"],
         "C": [1e-4, 1e-3, 1e-2, 1e-1, 0.5, 1., 5., 10., 15., 20., 25.,],

diff --git a/tpot/config/regressor_cuml.py b/tpot/config/regressor_cuml.py
@@ -46,15 +46,6 @@
         "normalize": [True, False]
     },
 
-    "cuml.ensemble.RandomForestRegressor": {
-        "n_estimators": [100, 300, 500,],
-        "split_algo": [0, 1,],
-        "max_depth": range(8, 20),
-        "max_features": np.arange(0.05, 1.01, 0.05),
-        "min_rows_per_node": range(2, 21),
-        "n_bins": [64,]
-    },
-
     "cuml.linear_model.Ridge": {
     },