Skip to content

Instantly share code, notes, and snippets.

@gabraganca
Created April 4, 2019 23:47
Show Gist options
  • Save gabraganca/e18b29cdd34f662fcef8da92029426ce to your computer and use it in GitHub Desktop.
Save gabraganca/e18b29cdd34f662fcef8da92029426ce to your computer and use it in GitHub Desktop.
Criar dataset pro surprise
reader = Reader(rating_scale=(0.5, 5))
data = Dataset.load_from_df(df_ratings[['userId', 'movieId', 'rating']],
reader)
full_trainset = data.build_full_trainset()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment