2017-03-18 15 views
0

내 친구가 만든 python 스크립트입니다.이 파일을 moviulz 데이터 세트에서 가져온 모든 영화 목록이 포함 된 내 django 프로젝트에 통합하는 방법.이 코드를 어디에 통합해야합니까?추천 시스템 (파이썬 파일)을 django 프로젝트에 통합하는 방법.

import numpy as np 
import pandas as pd 

# set some print options 
np.set_printoptions(precision=4) 
np.set_printoptions(threshold=5) 
np.set_printoptions(suppress=True) 
pd.set_option('precision', 3, 'notebook_repr_html', True,) 
# init random gen 
np.random.seed(2) 



#users_file = "/media/sourabhkondapaka/Sourabh's/main_project/sandbox/ml-latest-small/ratings.csv" 
#movies_file = "/media/sourabhkondapaka/Sourabh's/main_project/sandbox/ml-latest-small/movies.csv" 
#users = pd.read_table(users_file,sep=',', header=None,names = ['user_id','movie_id','rating','timestamp']) 
#movies = pd.read_table(movies_file, sep=',') 



class popularity_based(): 
    def __init__(self,users,movies): 
     self.users = users 
     self.movies = movies 
     self.user_id = None 
     self.mean_ratings = None 
     self.movielens= None 
     self.c = 0 

    def create(self): 
     self.movielens = pd.merge(users,movies) 
     self.movie_ratings = self.movielens.ix[:,1:3] 
     self.mean_ratings = self.movie_ratings.groupby('movie_id',as_index = True)['rating'].mean().sort_values(ascending = False) 
     self.mean_ratings = pd.DataFrame(self.mean_ratings).reset_index() 
     self.mean_ratings['title'] = self.mean_ratings['movie_id'].map(self.movies.set_index('movie_id')['title']) 

    def recommend(self, user_id,topu): #no arguement required here, just for the sake of uniformness across other recommender implementations 
     self.user_id = user_id 
     #From = self.c 
     #self.c += topu 
     #To = self.c 
     print(type(self.mean_ratings.as_matrix(columns=None))) 
     return self.mean_ratings.ix[:topu,'title'].as_matrix(columns = None) 

답변

0

이 파일의 기능은 무엇입니까? 사용자가 영화 추천을받는 데 도움이되는 유틸리티로 작동하는 경우 기본 프로젝트 아래의 utils.py에 포함되어야합니다. 적절한 상대 가져 오기를 사용하여 올바른보기에서 콘텐츠를 제공 할 수 있습니다.