모델 성능을 측정하기 위한 함수를 정의합니다.
코드 9-34 모델 성능 측정 함수 정의
def accuracy(y_pred, y_test):
y_pred_tag = torch.round(torch.sigmoid(y_pred))
correct_results_sum = (y_pred_tag == y_test).sum().float() ------ 실제 정답과 모델의 결과가 일치하는 개수를 실수 형태로 변수에 저장
acc = correct_results_sum/y_test.shape[0]
acc = torch.round(acc * 100) ------ ①
return acc
① torch.round()는 반올림을 할 때 사용하며 사용 방법은 다음과 같습니다.
print(round(3.2)) print(round(8.7)) print(round(-3.2))
반올림이 적용된 결과는 다음과 같습니다.
3 9 -3