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]