我是深度学习的新手,我想得到我的模型的训练(accuracy,瓦尔_accuracy)和测试(accuracy,val_accuracy),我还想知道如何使用混淆矩阵打印精度,召回率和f1得分值。
这是我的模型的一些细节
vgg_model = VGG16(weights='imagenet', include_top=False, input_shape=(224, 224, 3))
for layer in vgg_model.layers:
layer.trainable = False
my_vgg_model = Sequential()
my_vgg_model.add(vgg_model)
my_vgg_model.add(Flatten())
my_vgg_model.add(Dense(256, activation='relu'))
my_vgg_model.add(Dense(128, activation='relu'))
my_vgg_model.add(Dense(2, activation='softmax'))
my_vgg_model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
hist = my_vgg_model.fit(x_train, y_train,
batch_size=32,
epochs=1,
verbose=1,
validation_data=(x_test, y_test),
callbacks=[cp_callback])
我没有尝试过任何东西,因为我无法分别获得这些训练和测试的准确性:(
1条答案
按热度按时间yxyvkwin1#
训练后,训练集和测试集的损失和准确率存储在hist变量中。您可以通过调用
hist.history
来访问它们作为字典。对于其他指标,您可以使用
sklearn.metrics
,如下所示: