Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
dataset.rst 8.1 KB
Newer Older
VoVAllen's avatar
[Dataset] Add CoraFull, Amazon, KarateClub, Coauthor Dataset (#855)  
VoVAllen committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48 
Dataset (Temporary)


.. table:: 

    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |  Datset Name   |                          Usage                           |# of graphs|Avg. # of nodes|Avg. # of edges|                 Node field                 |Edge field |Temporal|
    +================+==========================================================+===========+===============+===============+============================================+===========+========+
    |BitcoinOTC      |BitcoinOTC()                                              |        136|        6005.00|       21209.98|                                            |h          |True    |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Cora            |CoraDataset()                                             |          1|        2708.00|       10556.00|train_mask, val_mask, test_mask, label, feat|           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Citeseer        |CitationGraphDataset('citeseer')                          |          1|        3327.00|        9228.00|train_mask, val_mask, test_mask, label, feat|           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |PubMed          |CitationGraphDataset('pubmed')                            |          1|       19717.00|       88651.00|train_mask, val_mask, test_mask, label, feat|           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |QM7b            |QM7b()                                                    |       7211|          15.42|         244.95|                                            |h          |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Reddit          |RedditDataset()                                           |          1|      232965.00|   114615892.00|train_mask, val_mask, test_mask, feat, label|           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |ENZYMES         |TUDataset('ENZYMES')                                      |        600|          32.63|         124.27|node_labels, node_attr                      |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |DD              |TUDataset('DD')                                           |       1178|         284.32|        1431.32|node_labels                                 |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |COLLAB          |TUDataset('COLLAB')                                       |       5000|          74.49|        9830.00|                                            |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |MUTAG           |TUDataset('MUTAG')                                        |        188|          17.93|          39.59|node_labels                                 |edge_labels|False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |PROTEINS        |TUDataset('PROTEINS')                                     |       1113|          39.06|         145.63|node_labels, node_attr                      |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |PPI             |PPIDataset('train')/PPIDataset('valid')/PPIDataset('test')|         20|        2245.30|       63563.70|feat                                        |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |KarateClub      |KarateClub()                                              |          1|          34.00|         156.00|label                                       |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Amazon computer |AmazonCoBuy('computers')                                  |          1|       13752.00|      574418.00|feat, label                                 |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Amazon photo    |AmazonCoBuy('photo')                                      |          1|        7650.00|      287326.00|feat, label                                 |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Coauthor cs     |Coauthor('cs')                                            |          1|       18333.00|      327576.00|feat, label                                 |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |Coauthor physics|Coauthor('physics')                                       |          1|       34493.00|      991848.00|feat, label                                 |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |GDELT           |GDELT('train')/GDELT('valid')/GDELT('test')               |       2304|       23033.00|      811333.15|                                            |rel_type   |True    |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |ICEWS18         |ICEWS18('train')/ICEWS18('valid')/ICEWS18('test')         |        240|       23033.00|      192640.22|                                            |rel_type   |True    |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+
    |CoraFull        |CoraFull()                                                |          1|       19793.00|      130622.00|feat, label                                 |           |False   |
    +----------------+----------------------------------------------------------+-----------+---------------+---------------+--------------------------------------------+-----------+--------+