update testcases and examples

VIDA-NYU · Jul 12, 2023 · f9ef217 · f9ef217
1 parent e93026c
commit f9ef217
Show file tree

Hide file tree

Showing 5 changed files with 1,343 additions and 15,452 deletions.
diff --git a/alpha_automl/hyperparameter_tuning/smac.py b/alpha_automl/hyperparameter_tuning/smac.py
@@ -79,7 +79,7 @@ def get_primitive_params(config, step_name):
     class_params = {}
     for param in params:
         class_params[param] = config[param]
-    logger.critical(f"[SMAC] {step_name}: {class_params}")
+    logger.debug(f"[SMAC] {step_name}: {class_params}")
     return class_params
 
 

diff --git a/alpha_automl/hyperparameter_tuning/smac_parameters.json b/alpha_automl/hyperparameter_tuning/smac_parameters.json
@@ -1,5 +1,15 @@
 {
-    "sklearn.impute.SimpleImputer": {},
+    "sklearn.impute.SimpleImputer": {
+        "strategy": {
+            "type": "Categorical",
+            "value": [
+                "mean",
+                "median",
+                "most_frequent"
+            ],
+            "default": "mean"
+        }
+    },
     "sklearn.compose.ColumnTransformer": {},
     "sklearn.preprocessing.MaxAbsScaler": {},
     "sklearn.preprocessing.RobustScaler": {},
@@ -400,7 +410,7 @@
             "type": "Integer",
             "value": [
                 1,
-                100
+                1000
             ],
             "default": 1
         }
@@ -419,9 +429,17 @@
             "type": "Integer",
             "value": [
                 1,
-                1000
+                10000
             ],
             "default": 100
+        },
+        "min_samples_split": {
+            "type": "Float",
+            "value": [
+                0,
+                1
+            ],
+            "default": 0.8
         }
     },
     "sklearn.neighbors.KNeighborsRegressor": {
@@ -537,15 +555,15 @@
             "type": "Integer",
             "value": [
                 3,
-                20
+                50
             ],
             "default": 3
         },
         "max_leaf_nodes": {
             "type": "Integer",
             "value": [
                 1,
-                100
+                10000
             ],
             "default": 3
         }