Movielens

Utilities for fetching the Movielens datasets 1.

References

1

https://grouplens.org/datasets/movielens/

spotlight.datasets.movielens.get_movielens_dataset(variant='100K')[source]

Download and return one of the Movielens datasets.

Parameters

variant (string, optional) – String specifying which of the Movielens datasets to download. One of (‘100K’, ‘1M’, ‘10M’, ‘20M’).

Returns

Interactions – instance of the interactions class

Return type

spotlight.interactions.Interactions