TensorFlow：在二进制分类的上下文中使用F1_score

Question

我正在使用softmax函数执行二进制分类任务。 我的测试标签是一个热门列表，看起来像：test_y = [[1。 0.] [1。 0.]…]

预测标签是概率列表：

test_y_pred = [[  4.39091297e-09   1.00000000e+00]
        [  1.75207238e-10   1.00000000e+00] …]

当我尝试使用f1_score时，出现错误：

ValueError：无法处理二进制和连续的混合

我该如何处理？

谢谢

Answer 1

f1_score不会为您分类结果。

将您的预测更改为类的向量，例如：

import numpy as np
test_y = [np.argmax(prediction) for prediction in test_y]
test_y_pred= [np.argmax(prediction) for prediction in test_y_pred]