mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-06 20:04:24 +08:00
2c6d071617
* [Model] MixHop for Node Classification task * [docs] update * [docs] update * [fix] remove seed option * [fix] update readme * [feature] support qm9 dataset * [style] update * [docs] fix the details * [fix] indexing only support int * [style] update * [fix] multiple backends support * [docs] add qm9 * [fix] Z type: float32 -> int32 * [fix] Z type: int32 -> long * [docs] add ref * [docs] fix * [docs] update * [docs] update * [fix] test eval * [docs] fix example Co-authored-by: xnuohz@126.com <ubuntu@ip-172-31-44-184.us-east-2.compute.internal> Co-authored-by: Mufei Li <mufeili1996@gmail.com> Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
220 lines
4.3 KiB
ReStructuredText
220 lines
4.3 KiB
ReStructuredText
.. _apidata:
|
|
|
|
dgl.data
|
|
=========
|
|
|
|
.. currentmodule:: dgl.data
|
|
.. automodule:: dgl.data
|
|
|
|
Quick links:
|
|
|
|
* `Node Prediction Datasets`_
|
|
* `Edge Prediction Datasets`_
|
|
* `Graph Prediction Datasets`_
|
|
|
|
Base Dataset Class
|
|
---------------------------
|
|
|
|
.. autoclass:: DGLDataset
|
|
:members: download, save, load, process, has_cache, __getitem__, __len__
|
|
|
|
|
|
Node Prediction Datasets
|
|
---------------------------------------
|
|
|
|
DGL hosted datasets for node classification/regression tasks.
|
|
|
|
.. _sstdata:
|
|
|
|
Stanford sentiment treebank dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: SSTDataset
|
|
:members: __getitem__, __len__
|
|
|
|
|
|
.. _karateclubdata:
|
|
|
|
Karate club dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: KarateClubDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _citationdata:
|
|
|
|
Citation network dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: CoraGraphDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: CiteseerGraphDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: PubmedGraphDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _corafulldata:
|
|
|
|
CoraFull dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: CoraFullDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _rdfdata:
|
|
|
|
RDF datasets
|
|
```````````````````````````````````
|
|
|
|
.. autoclass:: AIFBDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: MUTAGDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: BGSDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: AMDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _amazoncobuydata:
|
|
|
|
Amazon Co-Purchase dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: AmazonCoBuyComputerDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: AmazonCoBuyPhotoDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _coauthordata:
|
|
|
|
Coauthor dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: CoauthorCSDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: CoauthorPhysicsDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _ppidata:
|
|
|
|
Protein-Protein Interaction dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: PPIDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _redditdata:
|
|
|
|
Reddit dataset
|
|
``````````````
|
|
.. autoclass:: RedditDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _sbmdata:
|
|
|
|
Symmetric Stochastic Block Model Mixture dataset
|
|
````````````````````````````````````````````````
|
|
.. autoclass:: SBMMixtureDataset
|
|
:members: __getitem__, __len__, collate_fn
|
|
|
|
|
|
Edge Prediction Datasets
|
|
---------------------------------------
|
|
|
|
DGL hosted datasets for edge classification/regression and link prediction tasks.
|
|
|
|
.. _kgdata:
|
|
|
|
Knowlege graph dataset
|
|
```````````````````````````````````
|
|
|
|
.. autoclass:: FB15k237Dataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: FB15kDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: WN18Dataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _bitcoinotcdata:
|
|
|
|
BitcoinOTC dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: BitcoinOTCDataset
|
|
:members: __getitem__, __len__
|
|
|
|
ICEWS18 dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: ICEWS18Dataset
|
|
:members: __getitem__, __len__
|
|
|
|
GDELT dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: GDELTDataset
|
|
:members: __getitem__, __len__
|
|
|
|
|
|
Graph Prediction Datasets
|
|
---------------------------------------
|
|
|
|
DGL hosted datasets for graph classification/regression tasks.
|
|
|
|
|
|
.. _qm7bdata:
|
|
|
|
QM7b dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: QM7bDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _qm9data:
|
|
|
|
QM9 dataset
|
|
```````````````````````````````````
|
|
.. autoclass:: QM9Dataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _minigcdataset:
|
|
|
|
Mini graph classification dataset
|
|
`````````````````````````````````
|
|
.. autoclass:: MiniGCDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _tudata:
|
|
|
|
TU dataset
|
|
``````````
|
|
.. autoclass:: TUDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. autoclass:: LegacyTUDataset
|
|
:members: __getitem__, __len__
|
|
|
|
.. _gindataset:
|
|
|
|
Graph isomorphism network dataset
|
|
```````````````````````````````````
|
|
|
|
.. autoclass:: GINDataset
|
|
:members: __getitem__, __len__
|
|
|
|
Utilities
|
|
-----------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
utils.get_download_dir
|
|
utils.download
|
|
utils.check_sha1
|
|
utils.extract_archive
|
|
utils.split_dataset
|
|
utils.load_labels
|
|
utils.save_info
|
|
utils.load_info
|
|
|
|
.. autoclass:: dgl.data.utils.Subset
|
|
:members: __getitem__, __len__
|