Порядок мультиклассовых прогнозов в кератах - PullRequest
0 голосов
/ 08 мая 2018

Я обучил модель с несколькими метками (2) и несколькими классами (3) с помощью keras и model.predict (test_data) дает 3 (num_samples, 1) ndarray ... пока все хорошо, но как я могу Проверьте, какой массив прогнозирования для какого класса: Моя модель архитектуры:

Input - Embedding - GRU1 - BatchNorm - Dropout - GRU2 - Dense(1) for Class1 - Dense(1) for Class2 - Dense(1) for Class3 

Keras документы упоминают только то, что предикат () возвращает 'массив (ы) предсказаний Numpy', но ничего о сохранении порядка. Спасибо,

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...