diff --git a/problem_examples.ipynb b/problem_examples.ipynb index bcbbee3e..866a4346 100644 --- 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",