search

jliphard / DeepEvolve

Rapid hyperparameter discovery for neural nets using genetic algorithms
Other
180 stars 56 forks source link

Last dense layer in CNN uses fixed value #3

Closed pentageonate closed 6 years ago

pentageonate commented 6 years ago

Line 228 in train.py uses a fixed value rather than reading the length of the nb_neurons array.

Changing the line to the following may be helpful as a suggestion: model.add(Dense(nb_neurons[len(nb_neurons) - 1], activation = activation))

jliphard commented 6 years ago

Good catch - I made the change - thanks!