반도체 소자 이상 탐지 AI 경진대회

알고리즘 | 월간 데이콘 | 비전 | 비지도학습 | 이상 탐지 | F1 Score

 

Resnet-18 + autoencoder

2024.02.14 17:05 3,062 조회 language

이상탐지 문제여서 바로  autoencoder가 생각나서 짜본 코드 입니다.
점수는 epoch, 스케쥴러 layer 등에 따라 0.3~0.7 왔다갔다 하네요
실험해본다고 시행착오식으로 막 코드를 계획없이 짜서 해당 코드의 score는 정확히 모르겠어요 ㅜㅜ 

one-class SVM, isolation forest등 머신러닝 모델 보다 점수가 안나오는거 같지만 혹시나 도움이 될까 하고 올려봅니다 ㅎㅎ

남은 기간 모두 파이팅 합시다!!

코드
로그인이 필요합니다
0 / 1000
도비콘
2024.02.14 18:09

👏🏻