Welcome to rs_datasets

This tool allows you download, unpack and read recommender systems datasets into pandas.DataFrame as easy as data = Dataset().

Installation

pip install rs_datasets

Available datasets

The following datasets are available for automatic download and can be retrieved with this package.

Note

Check dataset license to know available usecases. Authors of this package are not affiliated with dataset contents in any way.

Dataset Users Items Interactions
Movielens 162k 62k up to 25m
Million Song Dataset 1m 385k 48m
Netflix 480k 17.7k 100m
Goodreads 800k 1.5m 225m
Last.fm 360k 290k 17.5m
Epinions 49k 140k 660k
Book Crossing 279k 271k 1.1m
Dating Agency 135k 169k 17.3m
Jester 73k 100 4.1m
Amazon ?1 ?1 up to 32m

Example of use

from rs_datasets import MovieLens
ml = MovieLens()
ml.info()
ratings
   user_id  item_id  rating  timestamp
0        1        1     4.0  964982703
1        1        3     4.0  964981247
2        1        6     4.0  964982224
items
   item_id  ...                                       genres
0        1  ...  Adventure|Animation|Children|Comedy|Fantasy
1        2  ...                   Adventure|Children|Fantasy
2        3  ...                               Comedy|Romance
[3 rows x 3 columns]
tags
   user_id  item_id              tag   timestamp
0        2    60756            funny  1445714994
1        2    60756  Highly quotable  1445714996
2        2    60756     will ferrell  1445714992
links
   item_id  imdb_id  tmdb_id
0        1   114709    862.0
1        2   113497   8844.0
2        3   113228  15602.0

Loaded DataFrames are available as class attributes.


  1. Their download speed is extremely slow and I wasn't patient enough to download the biggest one to check this.