Commit b71a8aa6 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Adding CGCNN and GAT docs

parent 9b4d2273

docs/models.rst

+11 −0
Original line number Diff line number Diff line
@@ -421,4 +421,15 @@ You can wrap an arbitrary :code:`torch.nn.Module` in a :code:`TorchModel` object 
.. autoclass:: deepchem.models.TorchModel

  :members:



CGCNNModel

----------



.. autoclass:: deepchem.models.CGCNNModel

  :members:





GATModel

--------



.. autoclass:: deepchem.models.GATModel

  :members:

docs/requirements.txt

+1 −0
Original line number Diff line number Diff line
@@ -4,3 +4,4 @@ sphinx_rtd_theme 
tensorflow==2.2.0

transformers

xgboost

torch==1.6.0

CRA Git | Maintained and supported by SUSTech CRA and CCSE