Мне интересно, что не так с моей матрицей путаницы?
Мой код:
test_generator = test_datagen.flow_from_directory(
test_dir,
target_size=(150, 150),
color_mode='rgb',
batch_size=1,
class_mode='binary')
predictions = model.predict_generator(test_generator, steps=int(4000))
predicted_classes = np.argmax(predictions, axis=1)
print("Confusion Matrix")
print(confusion_matrix(test_generator.classes, predicted_classes))
Все, что он возвращает, это:
Confusion Matrix
[[2000 0]
[2000 0]]