from fastai.gen_doc.nbdoc import *

from fastai import *
from fastai.collab import * 

path = untar_data(URLs.ML_SAMPLE)
ratings = pd.read_csv(path/'ratings.csv')
ratings.head()

data = CollabDataBunch.from_df(ratings)

learn = collab_learner(data, n_factors=50, y_range=(0.,5.))

learn.fit_one_cycle(5, 5e-3, wd=0.1)

show_doc(CollabDataBunch, doc_string=False)

show_doc(CollabDataBunch.from_df, doc_string=False)

show_doc(EmbeddingDotBias, doc_string=False, title_level=3)

show_doc(collab_learner, doc_string=False)

show_doc(CollabLine, doc_string=False, title_level=3)

show_doc(CollabList, title_level=3, doc_string=False)

show_doc(EmbeddingDotBias.forward)