플랫폼: 'Goolge Colab'
GPU: Tesla T4
GPU API: cuda
모델 : CatBoost
-
bin_classfier.pkl - 0과 0이 아닌 데이터들을 구분하기 위한 이진 분류 모델(피클 파일)
-
multi_classfier.pkl - 1,2,3,4,5,6 데이터들을 구분하기 위한 다중 분류 모델(피클 파일)
-
모델_학습.ipynb - 코랩 환경에서 구동 가능한 이진분류 모델, 다중 분류 모델 학습 코드, GPU 사용, catboost 설치 필요, 데이터셋 경로 설정 필요.
-
모텔_테스트.ipynb - 코랩 환경에서 구동 가능한 경로를 통해서 피클 모델 및 데이터 호출 후 평가 할 수 있는 코드, catboost 설치 필요, 데이터셋 경로 및 모델 경로 설정 필요.
-
아이디어 요약본 - EDA, 데이터 전처리 및 모델제작에 대한 아이디어 요약.
-
아이디어 설명 - 아이디어에 대한 자세한 설명과 세부사항 및 결과 수록.
|
👦🏻 정민준 (EDA, 모델 제작, 발표) |