MovieLens1M

class dhg.data.MovieLens1M(data_root=None)[source]

Bases: dhg.data.base.BaseData

The MovieLens1M dataset is collected for user-item recommendation task. Wherein, movies are viewed as items. Released 2/2003. The dataset contains 1 million ratings from 6022 users on 3043 items.

The content of the MovieLens-1M dataset includes the following:

  • num_users: The number of users: \(6,022\).

  • num_items: The number of items: \(3,043\).

  • num_interactions: The number of interactions: \(995,154\).

  • train_adj_list: The train adjacency list.

  • test_adj_list: The test adjacency list.

Note

The first item of each line in the adj_list is the user id, and the rest is the item id.

Parameters

data_root (str, optional) – The data_root has stored the data. If set to None, this function will auto-download from server and save into the default direction ~/.dhg/datasets/. Defaults to None.