F1-Score Pytorch 트레이닝 모델이 출력된 후 F1-Score 및 AUC 작업을 계산합니다. 두 가지 분류에 있어batchsize의 크기가 64라고 가정하면 모델 하나의batch의 출력은torch가 되어야 한다.size([64,2]), 그래서 우선 이 2차원 행렬의 각 줄의 최대 인덱스 값을 얻어서 하나의 목록에 추가하고, 라벨도 하나의 목록에 추가하고, 마지막에 sklearn에서 F1을 계산하는 도구 패키지를 사용하여 계산합니다. AUC를 계산할 때 이번 사용은 sklearn의 r... Pytorch훈련 모형F1-ScoreAUC
Pytorch 트레이닝 모델이 출력된 후 F1-Score 및 AUC 작업을 계산합니다. 두 가지 분류에 있어batchsize의 크기가 64라고 가정하면 모델 하나의batch의 출력은torch가 되어야 한다.size([64,2]), 그래서 우선 이 2차원 행렬의 각 줄의 최대 인덱스 값을 얻어서 하나의 목록에 추가하고, 라벨도 하나의 목록에 추가하고, 마지막에 sklearn에서 F1을 계산하는 도구 패키지를 사용하여 계산합니다. AUC를 계산할 때 이번 사용은 sklearn의 r... Pytorch훈련 모형F1-ScoreAUC