minor

VIDA-NYU · Jun 13, 2023 · b4581ca · b4581ca
1 parent 93c8960
commit b4581ca
Showing 1 changed file with 30 additions and 37 deletions.
diff --git a/alpha_automl/hyperparameter_tuning/smac_parameters.json b/alpha_automl/hyperparameter_tuning/smac_parameters.json
@@ -39,7 +39,32 @@
     },
     "sklearn.discriminant_analysis.LinearDiscriminantAnalysis": {},
     "sklearn.discriminant_analysis.QuadraticDiscriminantAnalysis": {},
-    "sklearn.ensemble.BaggingClassifier": {},
+    "sklearn.ensemble.BaggingClassifier": {
+        "n_estimators": {
+            "type": "Integer",
+            "value": [
+                200,
+                2000
+            ],
+            "default": 400
+        },
+        "max_samples": {
+            "type": "Float",
+            "value": [
+                0,
+                1
+            ],
+            "default": 0.5
+        },
+        "max_features": {
+            "type": "Float",
+            "value": [
+                0,
+                1
+            ],
+            "default": 0.5
+        }
+    },
     "sklearn.ensemble.ExtraTreesClassifier": {
         "random_state": {
             "type": "Integer",
@@ -50,33 +75,17 @@
             "default": 0
         },
         "n_estimators": {
-            "type": "Categorical",
+            "type": "Integer",
             "value": [
                 200,
-                400,
-                600,
-                800,
-                1000,
-                1200,
-                1400,
-                1600,
-                1800,
                 2000
             ],
             "default": 400
         },
         "max_depth": {
-            "type": "Categorical",
+            "type": "Integer",
             "value": [
                 10,
-                20,
-                30,
-                40,
-                50,
-                60,
-                70,
-                80,
-                90,
                 100
             ],
             "default": 40
@@ -85,33 +94,17 @@
     "sklearn.ensemble.GradientBoostingClassifier": {},
     "sklearn.ensemble.RandomForestClassifier": {
         "n_estimators": {
-            "type": "Categorical",
+            "type": "Integer",
             "value": [
                 200,
-                400,
-                600,
-                800,
-                1000,
-                1200,
-                1400,
-                1600,
-                1800,
                 2000
             ],
             "default": 400
         },
         "max_depth": {
-            "type": "Categorical",
+            "type": "Integer",
             "value": [
                 10,
-                20,
-                30,
-                40,
-                50,
-                60,
-                70,
-                80,
-                90,
                 100
             ],
             "default": 40