Матрица замешательства для кошек против собак - PullRequest
0 голосов
/ 29 марта 2020

Мне интересно, что не так с моей матрицей путаницы?

Мой код:

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]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...