Как использовать счет F1 в тензор2тенор

Question

Я хочу добавить оценку F1 к моей проблеме классификации мультикласса. Я не нашел F1 в списке встроенных метрик. F1 где-то в тензорном тензоре, или мне нужно переписать eval_metrics () и использовать мой код или взять его из TensorFlow?

Answer 1

Вы должны импортировать его из Scikit Learn:

from sklearn.metrics import f1_score

Тогда вы можете сделать

f1 = f1_score(y_test, y_classes)
print(f1)