from theano_bpr.utils import load_data_from_movielens

training_data, users_to_index, items_to_index = load_data_from_movielens('http://files.grouplens.org/datasets/movielens/ml-100k/ua.base', 3)

training_data[0:5]

users_to_index.keys()[0:10]

users_to_index['811']

items_to_index.keys()[0:10]

items_to_index['349']

testing_data, users_to_index, items_to_index = load_data_from_movielens('http://files.grouplens.org/datasets/movielens/ml-100k/ua.test', 3, users_to_index, items_to_index)

testing_data[0:5]

from theano_bpr import BPR

bpr = BPR(10, len(users_to_index.keys()), len(items_to_index.keys()))

bpr.train(training_data, epochs=50)

bpr.test(testing_data)