TensorFlow : 이진 분류 문맥에서 F1_score 사용

Question

이진 분류 작업에 softmax 함수를 사용하고 있습니다. 내 테스트 라벨은 하나의 핫리스트이며 다음과 같습니다 : test_y = [[1. 0] [1.

에 ValueError : 혼합을 처리 할 수 ​​

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

나는 f1_score를 사용하려고하면, 나는 오류가 발생 : 0] ...]

그리고 예측 라벨은 확률 목록입니다 이진 및 연속

이 문제는 어떻게 처리합니까?

감사

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]