GraphLab :: demo

hello.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import graphlab

graphlab.canvas.set_target('ipynb')

data = graphlab.SFrame.read_csv("http://s3.amazonaws.com/GraphLab-Datasets/movie_ratings/training_data.csv", column_type_hints={"rating":int})
data.head()

model = graphlab.recommender.create(data, user_id="user", item_id="movie", target="rating")

results = model.recommend(users=None, k=5)
results.head()

model.save("my_model")