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