使用tensorflow构建基础电影推荐系统

使用jupyter 可以方便调试

第一步：收集数据

https://grouplens.org/datasets/movielens/

第二步 准备数据

import pandas as pd
import numpy as np
import tensorflow as tf
ratings_df = pd.read_csv('ratings.csv')
ratings_df.tail()
movies_df = pd.read_csv('movies.csv')
movies_df['movieRow'] = movies_df.index #添加电影行数

筛选movies_df中的特征

movies_df = movies_df[['movieRow','movieId','title']]
movies_df.to_csv('moviesProcessed.csv',index = False,header = True,encoding = 'utf-8')

将ratings_df中的moviesId 替换成行号

ratings_df = pd.merge(ratings_df,movies_df,on = 'movieId')

筛选ratings_df中的特征

ratings_df = ratings_df[['userId','movieRow','rating']]
ratings_d