mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-05 19:54:25 +08:00
a3febc061b
* add gin model * convert dataset.py to data_ont_the_fly way and put it into dgl.data module * convert dataset.py to data_ont_the_fly way and put it into dgl.data module python code checked * modified document and reference TUDataset; checked python part and bypass cpp part due to error * change tensor to numpy in dataset and transform in collate@Dataloader * Change minor format issue Change minor format issue * moved logging; adjusted tqdm etc
60 lines
1.2 KiB
ReStructuredText
60 lines
1.2 KiB
ReStructuredText
.. _apidata:
|
|
|
|
Dataset
|
|
=======
|
|
|
|
.. currentmodule:: dgl.data
|
|
|
|
Utils
|
|
-----
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
utils.get_download_dir
|
|
utils.download
|
|
utils.check_sha1
|
|
utils.extract_archive
|
|
|
|
Dataset Classes
|
|
---------------
|
|
|
|
Stanford sentiment treebank dataset
|
|
```````````````````````````````````
|
|
|
|
For more information about the dataset, see `Sentiment Analysis <https://nlp.stanford.edu/sentiment/index.html>`__.
|
|
|
|
.. autoclass:: SST
|
|
:members: __getitem__, __len__
|
|
|
|
Mini graph classification dataset
|
|
`````````````````````````````````
|
|
|
|
.. autoclass:: MiniGCDataset
|
|
:members: __getitem__, __len__, num_classes
|
|
|
|
|
|
Graph kernel dataset
|
|
````````````````````
|
|
|
|
For more information about the dataset, see `Benchmark Data Sets for Graph Kernels <https://ls11-www.cs.tu-dortmund.de/staff/morris/graphkerneldatasets>`__.
|
|
|
|
.. autoclass:: TUDataset
|
|
:members: __getitem__, __len__
|
|
|
|
|
|
Graph isomorphism network dataset
|
|
```````````````````````````````````
|
|
|
|
A compact subset of graph kernel dataset
|
|
|
|
.. autoclass:: GINDataset
|
|
:members: __getitem__, __len__
|
|
|
|
|
|
Protein-Protein Interaction dataset
|
|
```````````````````````````````````
|
|
|
|
.. autoclass:: PPIDataset
|
|
:members: __getitem__, __len__
|