"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "fdcf368708d5e246df25dea5cf4e45a8b484a90f"
data.rst 2.39 KB
Newer Older
Mufei Li's avatar
Mufei Li committed
1
2
.. _apidata:

3
4
dgl.data
=========
Mufei Li's avatar
Mufei Li committed
5
6
7
8
9
10
11
12
13
14
15
16
17

.. currentmodule:: dgl.data

Utils
-----

.. autosummary::
    :toctree: ../../generated/

    utils.get_download_dir
    utils.download
    utils.check_sha1
    utils.extract_archive
18
    utils.split_dataset
VoVAllen's avatar
VoVAllen committed
19
20
21
    utils.save_graphs
    utils.load_graphs
    utils.load_labels
22
23
24

.. autoclass:: dgl.data.utils.Subset
    :members: __getitem__, __len__
Mufei Li's avatar
Mufei Li committed
25
26
27
28
29
30
31
32
33
34

Dataset Classes
---------------

Stanford sentiment treebank dataset
```````````````````````````````````

For more information about the dataset, see `Sentiment Analysis <https://nlp.stanford.edu/sentiment/index.html>`__.

.. autoclass:: SST
35
36
    :members: __getitem__, __len__

37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

Karate Club dataset
```````````````````````````````````

.. autoclass:: KarateClub
    :members: __getitem__, __len__


Citation Network dataset
```````````````````````````````````

.. autoclass:: CitationGraphDataset
    :members: __getitem__, __len__


CoraFull dataset
```````````````````````````````````

.. autoclass:: CoraFull
    :members: __getitem__, __len__


Amazon Co-Purchase dataset
```````````````````````````````````

.. autoclass:: AmazonCoBuy
    :members: __getitem__, __len__


Coauthor dataset
```````````````````````````````````

.. autoclass:: Coauthor
    :members: __getitem__, __len__


BitcoinOTC dataset
```````````````````````````````````

.. autoclass:: BitcoinOTC
    :members: __getitem__, __len__


ICEWS18 dataset
```````````````````````````````````

.. autoclass:: ICEWS18
    :members: __getitem__, __len__


QM7b dataset
```````````````````````````````````

.. autoclass:: QM7b
    :members: __getitem__, __len__



GDELT dataset
```````````````````````````````````

.. autoclass:: GDELT
    :members: __getitem__, __len__


102
103
104
Mini graph classification dataset
`````````````````````````````````

105
.. autoclass:: MiniGCDataset
106
    :members: __getitem__, __len__, num_classes
107

kitaev-chen's avatar
kitaev-chen committed
108

VoVAllen's avatar
VoVAllen committed
109
110
111
112
113
114
115
116
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__

kitaev-chen's avatar
kitaev-chen committed
117
118
119
120
121
122
123
124
125
126

Graph isomorphism network dataset
```````````````````````````````````

A compact subset of graph kernel dataset

.. autoclass:: GINDataset
    :members: __getitem__, __len__


127
128
129
130
131
Protein-Protein Interaction dataset
```````````````````````````````````

.. autoclass:: PPIDataset
    :members: __getitem__, __len__