fix NN input shape mismatch (#129)

mljar · Jul 21, 2020 · 8f2b756 · 8f2b756
1 parent 8cc0bd1
commit 8f2b756
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/supervised/tuner/hill_climbing.py b/supervised/tuner/hill_climbing.py
@@ -60,4 +60,9 @@ def get(params, ml_task, seed=1):
             params_2 = copy.deepcopy(params)
             params_2[key_to_update] = right
 
+        if "model_architecture_json" in params_1:
+            del params_1["model_architecture_json"]
+        if "model_architecture_json" in params_2:
+            del params_2["model_architecture_json"]
+
         return [params_1, params_2]