【发布时间】:2020-04-27 04:31:00
【问题描述】:
我正在使用 Keras ImageDataGenerator 类来加载、训练和预测。我已经尝试了here 的解决方案,但仍然有问题。我不确定我是否遇到与here 提到的相同的问题。我猜我的y_pred 和y_test 没有正确映射到彼此。
validation_generator = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='validation',
shuffle='False')
validation_generator2 = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='validation',
shuffle='False')
loss, acc = model.evaluate_generator(validation_generator,
steps=math.ceil(validation_generator.samples / batch_size),
verbose=0,
workers=1)
y_pred = model.predict_generator(validation_generator2,
steps=math.ceil(validation_generator2.samples / batch_size),
verbose=0,
workers=1)
y_pred = np.argmax(y_pred, axis=-1)
y_test = validation_generator2.classes[validation_generator2.index_array]
print('loss: ', loss, 'accuracy: ', acc) # loss: 0.47286026436090467 accuracy: 0.864
print('accuracy_score: ', accuracy_score(y_test, y_pred)) # accuracy_score: 0.095
来自 Keras 的 evaluate_generator 和来自 scikit learn 的 accuracy_score 给出了不同的准确度。当然,当我使用来自 scikit learn 的confusion_matrix(y_test, y_pred) 时,这给了我错误的混淆矩阵。我犯了什么错误? (y_test 我的意思是y_true)
更新:
为了显示y_test 和y_pred 不一致,我打印了每个类的准确性。
cm = confusion_matrix(y_test, y_pred)
cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]
cm.diagonal()
acc_each_class = cm.diagonal()
print('accuracy of each class: \n')
for i in range(len(labels)):
print(labels[i], ' : ', acc_each_class[i])
print('\n')
'''
accuracy of each class:
cannoli : 0.085
dumplings : 0.065
edamame : 0.1
falafel : 0.125
french_fries : 0.12
grilled_cheese_sandwich : 0.13
hot_dog : 0.075
seaweed_salad : 0.085
tacos : 0.105
takoyaki : 0.135
可以看出,每个类的准确率都太低了。
更新2:我如何训练模型,可能会有所帮助
train_generator = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='training')
validation_generator = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='validation',
shuffle='False')
validation_generator2 = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='validation',
shuffle='False')
loss = CategoricalCrossentropy()
model.compile(optimizer=SGD(lr=lr, momentum=momentum),
loss=loss,
metrics=['accuracy'])
history = model.fit_generator(train_generator,
steps_per_epoch = train_generator.samples // batch_size,
validation_data=validation_generator,
validation_steps=validation_generator.samples // batch_size,
epochs=epochs,
verbose=1,
callbacks=[csv_logger, checkpointer],
workers=12)
【问题讨论】:
标签: python tensorflow machine-learning keras scikit-learn