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) – Thedata_root
has stored the data. If set toNone
, this function will auto-download from server and save into the default direction~/.dhg/datasets/
. Defaults toNone
.