Результат прогноза в CNN - PullRequest
0 голосов
/ 24 мая 2018

Я работаю по следующей ссылке https://becominghuman.ai/building-an-image-classifier-using-deep-learning-in-python-totally-from-a-beginners-perspective-be8dbaf22dd8 Как мне узнать, что «один» в «результате [0] [0]» является предсказанием «собаки», а не кошки?

1 Ответ

0 голосов
/ 24 мая 2018

Код в статье использует класс ImageDataGenerator и функцию flow_from_directory() для загрузки и дополнения изображений.Проверяя документацию Keras , вы можете найти информацию об одном из аргументов, называемом classes:

Необязательный список подкаталогов классов (например, ['dogs', 'cats']),По умолчанию: нет.Если не указан, список классов будет автоматически выведен из имен / структуры подкаталога в каталоге, где каждый подкаталог будет обрабатываться как отдельный класс (а порядок классов, который будет сопоставлен с индексами меток, будет буквенно-цифровым).Словарь, содержащий сопоставление имен классов и индексов классов, можно получить с помощью атрибута class_indices.

На основании этой информации я подозреваю, что «кошка» относится к 0, тогда как «собака» относится к 1Чтобы убедиться, что вы можете проверить словарь классов:

print(train_datagen.class_indices)
...