Updated example.

hiive · Mar 10, 2021 · 03cd986 · 03cd986
1 parent 1023589
commit 03cd986
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/problem_examples.ipynb b/problem_examples.ipynb
@@ -939,7 +939,7 @@
     "y_test_hot = one_hot.transform(y_test.reshape(-1, 1)).todense()\n",
     "\n",
     "grid_search_parameters = {\n",
-    "    'max_iters': [5000],                          # nn params\n",
+    "    'max_iters': [1000],                          # nn params\n",
     "    'learning_rate': [1e-2],                       # nn params\n",
     "    'activation': [mlrose_hiive.relu],            # nn params\n",
     "    'restarts': [1],                             # rhc params\n",
@@ -953,7 +953,7 @@
     "    experiment_name='nn_test_rhc',\n",
     "    algorithm=mlrose_hiive.algorithms.rhc.random_hill_climb,\n",
     "    grid_search_parameters=grid_search_parameters,\n",
-    "    iteration_list=[1, 10, 50, 100, 250, 500, 1000, 2000, 5000],\n",
+    "    iteration_list=[1, 10, 50, 100, 250, 500, 1000],\n",
     "    hidden_layer_sizes=[[2]],\n",
     "    bias=True,\n",
     "    early_stopping=True,\n",