работы количество нейронов в каждом из слоёв рекомендуется использовать как в примере
ниже:
from keras.models import Sequential
from keras.layers import Dense
classifier = Sequential() # Инициализация НС
classifier.add(Dense(units = 16, activation = 'relu', input_dim = 30))
classifier.add(Dense(units = 8, activation = 'relu'))
classifier.add(Dense(units = 6, activation = 'relu'))
classifier.add(Dense(units = 1, activation = 'sigmoid'))
Укажите метод оптимизации
9
и функцию потери
10
:
classifier.compile(optimizer = 'rmsprop', loss = 'binary_crossentropy')
Теперь мы можем провести обучение нейронной сети. Вместо количества эпох, равного 100,
как в нижеследующем примере,
Достарыңызбен бөлісу: