Skip to content

Instantly share code, notes, and snippets.

@gabraganca

gabraganca/create_dataset.py

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.
Download ZIP
Criar dataset pro surprise
Raw
create_dataset.py
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