|
939 | 939 | "y_test_hot = one_hot.transform(y_test.reshape(-1, 1)).todense()\n",
|
940 | 940 | "\n",
|
941 | 941 | "grid_search_parameters = {\n",
|
942 |
| - " 'max_iters': [5000], # nn params\n", |
| 942 | + " 'max_iters': [1000], # nn params\n", |
943 | 943 | " 'learning_rate': [1e-2], # nn params\n",
|
944 | 944 | " 'activation': [mlrose_hiive.relu], # nn params\n",
|
945 | 945 | " 'restarts': [1], # rhc params\n",
|
|
953 | 953 | " experiment_name='nn_test_rhc',\n",
|
954 | 954 | " algorithm=mlrose_hiive.algorithms.rhc.random_hill_climb,\n",
|
955 | 955 | " 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", |
957 | 957 | " hidden_layer_sizes=[[2]],\n",
|
958 | 958 | " bias=True,\n",
|
959 | 959 | " early_stopping=True,\n",
|
|
0 commit comments