model.readout.rst 1.05 KB
Newer Older
1
2
.. _apimodelreadout:

Mufei Li's avatar
Mufei Li committed
3
4
Readout for Computing Graph Representations
===========================================
5

Mufei Li's avatar
Mufei Li committed
6
7
8
9
10
After updating node/edge representations with graph neural networks (GNNs), a common operation is to compute
graph representations out of updated node/edge representations. For example, we need to compute molecular
representations out of atom/bond representations in molecular property prediction. We call the various modules
for computing graph-level representations **readout** as in Neural Message Passing for Quantum Chemistry and this
section lists the readout modules implemented in DGL-LifeSci.
11

Mufei Li's avatar
Mufei Li committed
12
13
14
15
16
.. contents:: Contents
    :local:

AttentiveFP Readout
-------------------
17
18
19
.. automodule:: dgllife.model.readout.attentivefp_readout
    :members:

Mufei Li's avatar
Mufei Li committed
20
21
MLP Readout
-----------
22
23
24
.. automodule:: dgllife.model.readout.mlp_readout
    :members:

Mufei Li's avatar
Mufei Li committed
25
26
Weighted Sum and Max Readout
----------------------------
27
28
.. automodule:: dgllife.model.readout.weighted_sum_and_max
    :members:
29
30
31
32
33

Weave Readout
-------------
.. automodule:: dgllife.model.readout.weave_readout
    :members: