머신러닝1 내 데이터로 CNN 돌려보기 📌 캡스톤 프로젝트 데이터 준비하기 우리는 6개 클래스를 가지고 있고 어쩌구 근데 지금은 3개 클래스에 대한 데이터만 있어서 일단 세개로만 해보겠음. AWS Sagemaker를 이용할 계획인데, 그러려면 데이터셋을 pkl 파일로 만들어야 한다. 진짜 여기저기 삽질하다가 이 깃헙을 발견하고 따라해봤다! train, valid, test 폴더를 만들고 데이터를 7:2:1의 비율로 넣어주었다. 1. label이 들어간 csv 파일 만들어주기 csv 파일을 만들어주는 코드를 작성했다. import os import natsort import csv import re file_path = 'test/' file_lists = os.listdir(file_path) file_lists = natsort.natsor.. 2020. 9. 9. 이전 1 다음
