Показывать процент после прогноза в Керасе - PullRequest
0 голосов
/ 24 января 2019

Я тренировал модель с Keras, которая предсказывает теги в соответствии с постом.После обучения он проходит более 10 тестовых постов и предсказывает тег.Мне также нужно показывать точность прогноза для каждого поста, но я не знаю, что делать ... Это пример кода. Буду признателен за вашу помощь.

for i in range(10):
prediction = loaded_model.predict(np.array([x_test[i]]))
predicted_label = text_labels[np.argmax(prediction)]
print(test_posts.iloc[i][:50], "...")
print('Actual label:' + test_tags.iloc[i])
print("Predicted label: " + predicted_label + "\n")

1 Ответ

0 голосов
/ 25 января 2019

Я провел небольшой тест с некоторыми категориями, [0, 1, ..., 9]. Функция predict возвращает массив для вероятности каждого класса. Таким образом, вы можете просто напечатать max(prediction), если это категория и все, что вам нужно.

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