model.gnn.rst 946 Bytes
Newer Older
1
2
.. _apimodelgnn:

Mufei Li's avatar
Mufei Li committed
3
4
Graph Neural Networks for Updating Node/Edge Representations
============================================================
5

Mufei Li's avatar
Mufei Li committed
6
7
All models based on graph neural networks start with updating node/edge representations.
We introduce various GNN models implemented in DGL-LifeSci for representation update.
8

Mufei Li's avatar
Mufei Li committed
9
10
11
12
13
.. contents:: Contents
    :local:

AttentiveFP
-----------
14
15
16
.. automodule:: dgllife.model.gnn.attentivefp
    :members:

Mufei Li's avatar
Mufei Li committed
17
18
GAT
---
19
20
21
.. automodule:: dgllife.model.gnn.gat
    :members:

Mufei Li's avatar
Mufei Li committed
22
23
GCN
---
24
25
26
.. automodule:: dgllife.model.gnn.gcn
    :members:

Mufei Li's avatar
Mufei Li committed
27
28
MGCN
----
29
30
31
.. automodule:: dgllife.model.gnn.mgcn
    :members:

Mufei Li's avatar
Mufei Li committed
32
33
MPNN
----
34
35
36
.. automodule:: dgllife.model.gnn.mpnn
    :members:

Mufei Li's avatar
Mufei Li committed
37
38
SchNet
------
39
40
41
.. automodule:: dgllife.model.gnn.schnet
    :members:

42
43
44
45
46
Weave
-----
.. automodule:: dgllife.model.gnn.weave
    :members:

47
48
49
50
51
GIN
---
.. automodule:: dgllife.model.gnn.gin
    :members:

Mufei Li's avatar
Mufei Li committed
52
53
WLN
---
54
55
.. automodule:: dgllife.model.gnn.wln
    :members: