标签： tensor2tensor
我想在多类别分类问题上增加F1分数。我在内置指标列表中找不到F1。 F1是在tensor2tensor中的某个地方还是我必须覆盖eval_metrics（）并使用我的代码或从TensorFlow中获取它？
答案 0 :(得分：0)
您必须从scikit Learn导入它：
from sklearn.metrics import f1_score
那你就可以做
f1 = f1_score(y_test, y_classes) print(f1)