Keras中ANN每次迭代的权重值

时间:2018-11-05 00:32:39

标签: python neural-network keras

我正在尝试为每次迭代输出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 :(得分:0)