Skip to content

Commit 03cd986

Browse files
committed
Updated example.
1 parent 1023589 commit 03cd986

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

problem_examples.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -939,7 +939,7 @@
939939
"y_test_hot = one_hot.transform(y_test.reshape(-1, 1)).todense()\n",
940940
"\n",
941941
"grid_search_parameters = {\n",
942-
" 'max_iters': [5000], # nn params\n",
942+
" 'max_iters': [1000], # nn params\n",
943943
" 'learning_rate': [1e-2], # nn params\n",
944944
" 'activation': [mlrose_hiive.relu], # nn params\n",
945945
" 'restarts': [1], # rhc params\n",
@@ -953,7 +953,7 @@
953953
" experiment_name='nn_test_rhc',\n",
954954
" algorithm=mlrose_hiive.algorithms.rhc.random_hill_climb,\n",
955955
" grid_search_parameters=grid_search_parameters,\n",
956-
" iteration_list=[1, 10, 50, 100, 250, 500, 1000, 2000, 5000],\n",
956+
" iteration_list=[1, 10, 50, 100, 250, 500, 1000],\n",
957957
" hidden_layer_sizes=[[2]],\n",
958958
" bias=True,\n",
959959
" early_stopping=True,\n",

0 commit comments

Comments
 (0)