[python] 추천시스템 - surprise

2021. 5. 22. 22:32Python/문법

Ver. Jupyter Notebook (Anaconda3)

▶ surprise 패키지

   - surpriselib.com

   - 쉽게 사용 가능

   - Visual studio build tools 설치 되어 있어야 함

   - 설치: cmd > pip install scikit-surprise

 

 ● 추천 수행 프로세스

   - 데이터 로딩

     · 데이터 컬럼 (무조건 userid, itemid, rating 순서로 데이터 정렬)

     · DataFrame에서 데이터 로딩

   - 모델 설정 및 학습

     · 추천 Algorithm 설정 (SVD, KNNBasic 등)

     · Train 데이터로 학습 (train() 메소드)

   - 예측 및 평가

     · 예측 (test(): 전체 데이터 / predict(): 한개의 데이터)

     · 평가 (accuracy.rmse 등)