def run_examples():
	import random
	user = USER_DICT.keys()[random.randint(0, len(USER_DICT) - 1)]
	print user
	top = recommendations.top_matches(USER_DICT, user)
	print top
	recs = recommendations.get_recommendations(USER_DICT, user)[:10]
	print recs
	url = recs[0][1]
	more_top = recommendations.top_matches(recommendations.transform_prefs(USER_DICT), url)
	print more_top
예제 #2
0
def run_examples():
    import random
    user = USER_DICT.keys()[random.randint(0, len(USER_DICT) - 1)]
    print user
    top = recommendations.top_matches(USER_DICT, user)
    print top
    recs = recommendations.get_recommendations(USER_DICT, user)[:10]
    print recs
    url = recs[0][1]
    more_top = recommendations.top_matches(
        recommendations.transform_prefs(USER_DICT), url)
    print more_top
예제 #3
0
파일: 10_9.py 프로젝트: erikaris/cs834-f16
import os
import pickle
import random

import recommendations as r

if __name__ == '__main__':
    _, _, prefs = pickle.load(open('movielens.pkl'))

    # To make item-based, transform prefs, so the format of prefs will be prefs[movie][user] = rating
    prefs = r.transform_prefs(prefs)

    # Split prefs into train and test prefs
    movies = prefs.keys()
    random.shuffle(movies)

    movies_train, movies_test = movies[:int(0.9 * len(movies))], movies[int(0.1 * len(movies)):]
    train = {m: prefs[m] for m in movies_train}
    test = {m: prefs[m] for m in movies_test}

    for movie in test:
        sim_distances = []
        sim_pearsons = []

        for other_movie in train:
            # Calculate distance using euclidean distance
            sim_distances.append((r.sim_distance(prefs, movie, other_movie), other_movie))
            # Calculate similarity using pearson
            sim_pearsons.append((r.sim_pearson(prefs, movie, other_movie), other_movie))

        # distance sort ascending
예제 #4
0
# Library imports
import recommendations, deliciousrec

# Initialize delicious users
delusers=deliciousrec.initialize_user_dict('programming')
# Add myself to the dataset
delusers['rogerfernandezg']={}
# Fills delicious users with data from delicious
deliciousrec.fill_items(delusers)
# Show recommendations for specific user
user=delusers.keys()[1]
print recommendations.top_matches(delusers,user)[0:10]
url=recommendations.get_recommendations(delusers,user)[0][1]
print recommendations.top_matches(recommendations.transform_prefs(delusers),url)