Skip to content

Commit bf49c80

Browse files
author
Daniel
committed
neural network max_iter is now always 200
1 parent 12e6600 commit bf49c80

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

MicroLIA/optimization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -483,7 +483,7 @@ def hyper_opt(data_x=None, data_y=None, clf='rf', n_iter=25, opt_cv=None, balanc
483483
layers = tuple(params[layer] for layer in layers)
484484
model = MLPClassifier(hidden_layer_sizes=tuple(layers), learning_rate_init=params['learning_rate_init'],
485485
activation=params['activation'], learning_rate=params['learning_rate'], alpha=params['alpha'],
486-
batch_size='auto', solver=params['solver'], max_iter=2500, random_state=1909)
486+
batch_size='auto', solver=params['solver'], max_iter=200, random_state=1909)
487487
except:
488488
print('Failed to optimize with Optuna, switching over to BayesSearchCV...')
489489
params = {

0 commit comments

Comments
 (0)