[6기] 프로그래머스 인공지능 데브코스 111일차 TIL

1220

[17주차 - Day3] Recommendation system

무비렌즈 데이터를 활용한 아이템 협업 필터링

# 데이터 로딩 및 분석 파트

!pip install surprise

!wget "https://grepp-reco-test.s3.ap-northeast-2.amazonaws.com/movielens/movies.csv"
!wget "https://grepp-reco-test.s3.ap-northeast-2.amazonaws.com/movielens/ratings.csv"

- surprise 모듈을 이용할 때는 인터넷을 통한 데이터 사용이 불가능해서 직접 다운로드를 받아야 한다

 

from surprise import Dataset
from surprise import Reader
from collections import defaultdict
import numpy as np
import pandas as pd

movies = pd.read_csv("movies.csv")
ratings = pd.read_csv("ratings.csv")

movies.head()
ratings.head()

 

movie_ratings = pd.merge(movies, ratings, left_on='movieId', right_on='movieId')
movie_ratings.head()