lenskit.data.movielens#

Code to import MovieLens data sets into LensKit.

Attributes#

LOC

Classes#

MLData

Internal class representing an open ML data set.

ML100KLoader

Loader for the ML100K data set.

MLMLoader

Loader for the ML 1M and 10M data sets.

MLModernLoader

Loader for modern MovieLens data sets (20M and later).

Functions#

load_movielens(path)

Load a MovieLens dataset. The appropriate MovieLens format is detected

load_movielens_df(path)

Load the ratings from a MovieLens dataset as a raw data frame. The

Module Contents#

type lenskit.data.movielens.LOC = Path | tuple[ZipFile, str]#
lenskit.data.movielens.load_movielens(path)#

Load a MovieLens dataset. The appropriate MovieLens format is detected based on the file contents.

Stability:
Caller (see Stability Levels).
Parameters:

path (str | pathlib.Path) – The path to the dataset, either as an unpacked directory or a zip file.

Returns:

The dataset.

Return type:

lenskit.data._dataset.Dataset

lenskit.data.movielens.load_movielens_df(path)#

Load the ratings from a MovieLens dataset as a raw data frame. The appropriate MovieLens format is detected based on the file contents.

Stability:
Caller (see Stability Levels).
Parameters:

path (str | pathlib.Path) – The path to the dataset, either as an unpacked directory or a zip file.

Returns:

The ratings, with columns user_id, item_id, rating, and timestamp.

Return type:

pandas.DataFrame

Exported Aliases#

lenskit.data.movielens.get_logger()#

Re-exported alias for lenskit.logging.get_logger().

class lenskit.data.movielens.DatasetBuilder#

Re-exported alias for lenskit.data._builder.DatasetBuilder.

class lenskit.data.movielens.Dataset#

Re-exported alias for lenskit.data._dataset.Dataset.