k-fold, stratified-fold 실험 코드, 결과 공유

[nstalways]

변경 사항

1. architecture/model.py에 resnet50, myensemble 클래스가 추가되었습니다.
2. train.py에 fold 방식으로 학습할 수 있도록 train_with_fold라는 함수가 추가되었습니다. 사용법은 다음과 같습니다. on terminal) python train.py --mode=k # mode를 설정하지 않으면 원래 train함수가 호출됩니다. mode는 k(k-fold), s(stratified-fold), g(group fold)가 있으며, k과 s까지는 정상 동작합니다. s의 경우 일단 코드를 작성하긴 했는데, 실제 원하는 방식대로 동작하는지 검증해보지는 않아서 사용하실 때 주의가 필요합니다.

[97DongHyeokOH]

확인 했습니다. 저도 oversampling한 데이터 셋 제출 한 번 해보고 성능이 괜찮게 나온다면 'mode'에 적용할 수 있도록 코드 변경해보겠습니다.

[codehyunn]

확인했습니다! 실행하기 편하게 코드 구성해주셨네요👍

[nstalways]

리뷰 감사합니다. 피어세션 때 같이 merge하면 될 것 같네요.