def print_question(data, labels, user_ids, data_dicts):
option = 0
movies, genres = get_movie_genres(data, labels)
while(option != 2 and option != 1):
print ('Are you registered in the system? \n1 for Yes \n2 for No')
try:
option = int(input())
except:
print 'Please try Again, we could not understand the input'
continue
if option == 1:
new_id = 0
while not new_id in user_ids:
print ('Enter id assigned: ')
try:
new_id = int(input())
except:
print 'Please Enter a correct Id'
continue
recomendations, data_dicts = existing_user(movies=movies, user_id=new_id,
ranks=data_dicts[new_id].keys(),
data_dicts=data_dicts)
else:
new_id = max(user_ids) + 1
recomendations, data_dicts = existing_user(movies=movies, user_id=new_id,
data_dicts=data_dicts)
print 'We recomend the next movies for you:'
for x in recomendations[:5]:
print x
from read_data import read_file, get_ratexuser, get_movie_genres from similarity import recommend from user_personalization import print_question data, labels = read_file() # Get all the movie titles just one time movies, genres = get_movie_genres(data, labels) # separate the users by Id and then take just the ranking for each movie data_dicts, user_ids = get_ratexuser(data) print_question(data, labels, user_ids, data_dicts)
