1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch로 배우는 딥러닝 입문
Connected

演習

torchmetrics로 정확도 계산하기

훈련 중 정확도를 추적하면 가장 성능이 좋은 epoch을 확인할 수 있어요.

이 연습에서는 torchmetrics를 사용해 클래스가 세 개인 마스크 착용(facemask) 데이터셋에서 정확도를 계산해 봅니다. plot_errors 함수는 잘못 분류된 샘플을 강조 표시해, 모델의 오류를 분석하는 데 도움을 줍니다.

torchmetrics 패키지는 이미 임포트되어 있어요. 모델 outputs는 softmax 확률이며, labels는 원-핫(one-hot) 인코딩 벡터입니다.

指示1 / 2

undefined XP
    1
    2
  • 클래스가 세 개인 "multiclass" 정확도 메트릭을 만드세요.
  • 데이터로더의 각 배치에 대해 정확도를 계산하세요.