[Pytorch] Pnemonia Prediction Model – Prediction[Pytorch] Pnemonia Prediction Model – Prediction
이제 학습한 모델을 불러들이고 실제 테스트셋을 입력할 경우 Classification으로 정상인지 폐렴인지 학습을 잘해노았는지 검증하여보자. 모델 예측: 폐렴
[일:] 2024년 01월 13일
[Pytorch] Pnemonia Prediction Model – Train[Pytorch] Pnemonia Prediction Model – Train
Pytorch를 이용하여 사내에서 학습을 해야하는 환경이 있어 GPU환경이 잘 설정되었는지 테스트겸 Kaggle에서 데이터셋을 다운로드 받고 학습을 돌렸다. 오랜만에 AI 쪽 업무를 하려니 많이 까먹기도 재밌어지기도 했다. 데이터셋 리뷰 https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia Kaggle의