한국외국어대학교 AI교육원 2022 인공지능 아이디어 페스티벌
data.py
에 있는 구글 play store에 있는 리뷰 데이터를 전처리한다.
1차 전처리가 된dataset.csv
를 갖고 코드를 돌리면 된다.- 현재 더 많은 데이터로 학습을 시키기 위해 영어버전으로 학습을 시킨 후 데모를 진행하였다.
- 한국어 버전을 원하면
data/korean.py
에 한국어 playstore 크롤링 버전이 있으니 이를 이용하여 데이터셋을 만들고 아래 단계를 진행하면 된다.
- 데이터 세부 전처리 및 모델학습은 아래코드로 실행 가능하다
python3 model_train.py
python3 model_train.py
모델데모는 demo.ipynb
에서 확인 가능하다.