mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-07 20:41:20 +08:00
62 lines
967 B
ReStructuredText
62 lines
967 B
ReStructuredText
.. _apimodelzoo:
|
|
|
|
Model Zoo
|
|
=========
|
|
|
|
.. currentmodule:: dgl.model_zoo
|
|
|
|
Chemistry
|
|
---------
|
|
|
|
Utils
|
|
`````
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
chem.load_pretrained
|
|
|
|
Property Prediction
|
|
```````````````````
|
|
|
|
Currently supported model architectures:
|
|
|
|
* GCNClassifier
|
|
* GATClassifier
|
|
* MPNN
|
|
* SchNet
|
|
* MGCN
|
|
* AttentiveFP
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.GCNClassifier
|
|
:members: forward
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.GATClassifier
|
|
:members: forward
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.MPNNModel
|
|
:members: forward
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.SchNet
|
|
:members: forward
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.MGCNModel
|
|
:members: forward
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.AttentiveFP
|
|
:members: forward
|
|
|
|
Generative Models
|
|
`````````````````
|
|
|
|
Currently supported model architectures:
|
|
|
|
* DGMG
|
|
* JTNN
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.DGMG
|
|
:members: forward
|
|
|
|
.. autoclass:: dgl.model_zoo.chem.DGLJTNNVAE
|
|
:members: forward
|