from fastai import *          # Quick access to most common functionality
from fastai.collab import *   # Quick access to collab filtering functionality

path = untar_data(URLs.ML_SAMPLE)
path

ratings = pd.read_csv(path/'ratings.csv')
series2cat(ratings, 'userId', 'movieId')
ratings.head()

learn = get_collab_learner(ratings, n_factors=50, min_score=0., max_score=5.)
learn.fit_one_cycle(4, 5e-3)