작성한 질문수
최신 딥러닝 기술 Vision Transformer 개념부터 Pytorch 구현까지
작성
·
212
0
답변 1
안녕하세요!
accuracy 구하는 함수를 복사 하셔서 불필요한 것 지우시고 def confusion_matrix(....) 함수를 만들어주면 좋을 것 같아요 :)
모델에서 나온 아웃풋과 실제 라벨을 넘파이 concat이나, 리스트 append를 이용하여 각각 저장 한 다음에
모은 예측값과 실제값을 가지고 confusion matrix 코드에 바로 적용하시면 되요!!
해결하였습니다 감사합니다! 많이 배웠습니다.
해결하였습니다 감사합니다! 많이 배웠습니다.