dgl.data.rst 3.9 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

.. currentmodule:: dgl.data

8
9
Dataset Classes
---------------
Mufei Li's avatar
Mufei Li committed
10

11
12
DGL dataset
```````````
Mufei Li's avatar
Mufei Li committed
13

14
15
.. autoclass:: DGLDataset
    :members: download, save, load, process, has_cache, __getitem__, __len__
16

17
18
DGL builtin dataset
```````````````````
Mufei Li's avatar
Mufei Li committed
19

20
21
.. autoclass:: DGLBuiltinDataset
    :members: download
Mufei Li's avatar
Mufei Li committed
22

23
24
.. _sstdata:

Mufei Li's avatar
Mufei Li committed
25
26
27
28
29
Stanford sentiment treebank dataset
```````````````````````````````````

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

30
.. autoclass:: SSTDataset
31
32
    :members: __getitem__, __len__

33
.. _karateclubdata:
34

35
Karate club dataset
36
37
```````````````````````````````````

38
.. autoclass:: KarateClubDataset
39
40
    :members: __getitem__, __len__

41
.. _citationdata:
42

43
Citation network dataset
44
45
```````````````````````````````````

46
47
48
49
.. autoclass:: CoraGraphDataset
    :members: __getitem__, __len__

.. autoclass:: CiteseerGraphDataset
50
51
    :members: __getitem__, __len__

52
53
54
.. autoclass:: PubmedGraphDataset
    :members: __getitem__, __len__

55
.. _kgdata:
56
57
58
59
60
61
62
63
64
65
66
67
68

Knowlege graph dataset
```````````````````````````````````

.. autoclass:: FB15k237Dataset
    :members: __getitem__, __len__

.. autoclass:: FB15kDataset
    :members: __getitem__, __len__

.. autoclass:: WN18Dataset
    :members: __getitem__, __len__

69
.. _rdfdata:
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

RDF datasets
```````````````````````````````````

.. autoclass:: AIFBDataset
    :members: __getitem__, __len__

.. autoclass:: MUTAGDataset
    :members: __getitem__, __len__

.. autoclass:: BGSDataset
    :members: __getitem__, __len__

.. autoclass:: AMDataset
    :members: __getitem__, __len__

86
.. _corafulldata:
87
88
89
90

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

91
.. autoclass:: CoraFullDataset
92
93
    :members: __getitem__, __len__

94
.. _amazoncobuydata:
95
96
97
98

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

99
100
101
102
.. autoclass:: AmazonCoBuyComputerDataset
    :members: __getitem__, __len__

.. autoclass:: AmazonCoBuyPhotoDataset
103
104
    :members: __getitem__, __len__

105
.. _coauthordata:
106
107
108
109

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

110
111
112
113
.. autoclass:: CoauthorCSDataset
    :members: __getitem__, __len__

.. autoclass:: CoauthorPhysicsDataset
114
115
    :members: __getitem__, __len__

116
.. _bitcoinotcdata:
117
118
119
120

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

121
.. autoclass:: BitcoinOTCDataset
122
123
124
125
126
127
    :members: __getitem__, __len__


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

128
.. autoclass:: ICEWS18Dataset
129
130
    :members: __getitem__, __len__

131
.. _qm7bdata:
132
133
134
135

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

136
.. autoclass:: QM7bDataset
137
138
139
140
141
142
143
    :members: __getitem__, __len__



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

144
.. autoclass:: GDELTDataset
145
146
    :members: __getitem__, __len__

147
.. _minigcdataset:
148

149
150
151
Mini graph classification dataset
`````````````````````````````````

152
.. autoclass:: MiniGCDataset
153
    :members: __getitem__, __len__
VoVAllen's avatar
VoVAllen committed
154

155
156
.. _tudata:

157
158
TU dataset
``````````
VoVAllen's avatar
VoVAllen committed
159
160
161
162

.. autoclass:: TUDataset
    :members: __getitem__, __len__

163
164
.. autoclass:: LegacyTUDataset
    :members: __getitem__, __len__
kitaev-chen's avatar
kitaev-chen committed
165

166
167
.. _gindataset:

kitaev-chen's avatar
kitaev-chen committed
168
169
170
171
172
173
174
175
Graph isomorphism network dataset
```````````````````````````````````

A compact subset of graph kernel dataset

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

176
.. _ppidata:
kitaev-chen's avatar
kitaev-chen committed
177

178
179
180
181
182
Protein-Protein Interaction dataset
```````````````````````````````````

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

184
.. _redditdata:
185
186
187
188
189
190
191

Reddit dataset
``````````````

.. autoclass:: RedditDataset
    :members: __getitem__, __len__

192
.. _sbmdata:
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211

Symmetric Stochastic Block Model Mixture dataset
````````````````````````````````````````````````

.. autoclass:: SBMMixtureDataset
    :members: __getitem__, __len__, collate_fn

Utils
-----

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

    utils.get_download_dir
    utils.download
    utils.check_sha1
    utils.extract_archive
    utils.split_dataset
    utils.load_labels
212
213
    utils.save_info
    utils.load_info
214
215
216
217

.. autoclass:: dgl.data.utils.Subset
    :members: __getitem__, __len__