mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-05 19:54:25 +08:00
00fc680aae
* add graph classification dataset * add node label * add TUDataset * Modify to consistent with Qi Huang's implementation * add docs * Add docs * Fix change of environment variable * Update tu.py * Update tu.py * Fix error when add node with np.int64
49 lines
1.0 KiB
ReStructuredText
49 lines
1.0 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__
|
|
|
|
Protein-Protein Interaction dataset
|
|
```````````````````````````````````
|
|
|
|
.. autoclass:: PPIDataset
|
|
:members: __getitem__, __len__
|