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

1222

[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"

import numpy as np
import pandas as pd

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

movies.head()

 

ratings.head()

 

itemRatings = ratings.pivot_table(index=['userId'], columns=['movieId'], values='rating')
itemRatings.head()

 

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