Skip to content

Instantly share code, notes, and snippets.

@gabraganca

gabraganca/load_dataset.py

Created April 4, 2019 23:45
Show Gist options
  • Save gabraganca/6e5f1e1f2f1955823ef7a15303e1c248 to your computer and use it in GitHub Desktop.
Save gabraganca/6e5f1e1f2f1955823ef7a15303e1c248 to your computer and use it in GitHub Desktop.
Download ZIP
Load dataset
Raw
load_dataset.py
dataset = 'ml-latest-small'
# Carrega as notas
df_ratings = pd.read_csv(f'{dataset}/ratings.csv',
dtype={'userId':str, 'movieId':str, 'rating':float})
df_ratings.drop('timestamp', axis=1, inplace=True)
# Carrega informação dos filmes:
df_movies = pd.read_csv(f'{dataset}/movies.csv', dtype=str)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment