#tensor2tensor
#tensor2tensor
Вопрос:
Я хочу добавить оценку F1 к моей задаче мультиклассовой классификации. Я не нашел F1 в списке встроенных показателей. Есть ли F1 где-нибудь в tensor2tensor, или я должен перезаписать eval_metrics () и использовать свой код или взять его из TensorFlow?
Ответ №1:
Вы должны импортировать ее из scikit learn:
from sklearn.metrics import f1_score
Затем вы можете сделать
f1 = f1_score(y_test, y_classes)
print(f1)
Комментарии:
1. это не систематическое решение, должен быть какой-то более элегантный способ добавления «пользовательских» показателей в эту конкретную структуру