dmlc / dmlc/dgl

Several issues about TUDataset

Open
#2,906 2 comments 1 reaction 1 assignee Claimed by @VoVAllen View on GitHub
feature request
Dominant language
Python
Stars
14.3k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

I encountered several usability issues when developing the benchmark code on TUDataset.

1. There are two datasets `LegacyTUDataset` and `TUDataset`. They are quite different -- `TUDataset` does not have node features while `LegacyTUDataset` does. We should merge them and provide options to include node features per user request.
2. Does not support slicing or list indexing, which is very useful during train/val/test split. Ideally, `dataset[index_list]` should return a sub-dataset. I imagine this might be a common issue for all graph classification datasets.
3. Does not support shuffling. We should have sth like `dataset.shuffle()` which returns a new shuffled dataset, which is useful for creating stratified train/val/test splits.
4. `dataset.num_labels` returns a `numpy.ndarray` instead of an integer.
5. Lack an API for getting the number of input features, e.g., `dataset.num_features`.

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.