взвешивает значения за итерацию для ANN в Керасе - PullRequest
0 голосов
/ 05 ноября 2018

Я пытаюсь вывести весовые значения для последнего слоя моего ANN за одну итерацию. Но из следующего кода, который я написал, я мог вывести его только для последней итерации, что означает конечный результат. Так как я могу добавить его к каждой итерации?

model=Sequential()
model.add(Dense(3,activation='sigmoid',input_dim=8))
model.add(Dense(3,activation='sigmoid'))
model.add(Dense(10,activation='softmax'))

sgd=SGD(lr=0.1)
model.compile(optimizer=sgd,loss='categorical_crossentropy',metrics=['accuracy'])

print("Training----------")
model.fit(features_train,class_train,validation_data=(features_test,class_test),nb_epoch=100,batch_size=1)

weights = np.array(model.layers[2].get_weights())
print(weights)

1 Ответ

0 голосов
/ 05 ноября 2018

Я настоятельно рекомендую вам прочитать это:

https://machinelearningmastery.com/check-point-deep-learning-models-keras/

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...