Skip to content

Commit 917665f

Browse files
MovieLens Datasets (#147)
* MovieLens 100k * MovieLens 100k modular with test * MovieLens 1m + doc_error_fix + indentation fix * 20m+25m - tests - consistency * tests and better metadata * fix expected type * ml-10m * Docs * Don't run ogbn-mag on Windows + remove comment * Fix windows issue * Update src/datasets/graphs/movielens.jl * Update src/datasets/graphs/movielens.jl * Update src/datasets/graphs/movielens.jl * Update src/datasets/graphs/movielens.jl * Update src/datasets/graphs/movielens.jl * Update src/datasets/graphs/movielens.jl Co-authored-by: Carlo Lucibello <[email protected]>
1 parent b389708 commit 917665f

File tree

5 files changed

+794
-20
lines changed

5 files changed

+794
-20
lines changed

docs/src/datasets/graphs.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,5 @@ PolBlogs
2828
PubMed
2929
Reddit
3030
TUDataset
31+
MovieLens
3132
```

src/MLDatasets.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,8 @@ include("datasets/graphs/karateclub.jl")
118118
export KarateClub
119119
include("datasets/graphs/reddit.jl")
120120
export Reddit
121+
include("datasets/graphs/movielens.jl")
122+
export MovieLens
121123

122124
function __init__()
123125
# TODO automatically find and execute __init__xxx functions
@@ -130,6 +132,7 @@ function __init__()
130132
__init__pubmed()
131133
__init__reddit()
132134
__init__tudataset()
135+
__init__movielens()
133136

134137
# misc
135138
__init__iris()

0 commit comments

Comments
 (0)