dmlc / dmlc/dgl

[Feature Request] MovieLens dataset

Open
#4,542 0 comments 0 reactions 1 assignee Claimed by @rudongyu View on GitHub
feature request topic: GNN Utilities
Dominant language
Python
Stars
14.3k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature

Add a new `MovieLensDataset` to `dgl.data`.

## Motivation

Requested by https://discuss.dgl.ai/t/movielens-hetrogenous-graph-graphsage-link-prediction/3201/2

## Alternatives

Currently, the MovieLens dataset is available at the [GCMC example](https://github.com/dmlc/dgl/blob/master/examples/pytorch/gcmc/data.py). Users can copy the code but having a built-in module in DGL main library will be more convenient.

## Pitch

A new `dgl.data.MoveLensDataset` class that follows the standard interface of DGLDataset. It should support all three variants (i.e., ml-100k, ml-1m, ml-10m).

## Additional context

None

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.