Skip to content

GitLab

"tests/nn/vscode:/vscode.git/clone" did not exist on "7a3199b197568572063c0ed7f7e9f889ba3d284f"
Switch branch/tag
Find file Normal viewHistoryPermalink
api.rst 1.55 KB
Newer Older
Gao, Xiang's avatar
Add tutorials (#86)  
Gao, Xiang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 
TorchANI
========

.. autoclass:: torchani.AEVComputer
    :members:
.. autoclass:: torchani.ANIModel
.. autoclass:: torchani.Ensemble
.. autoclass:: torchani.EnergyShifter
    :members:
.. autoclass:: torchani.nn.Gaussian


Datasets
========

.. automodule:: torchani.data
.. autoclass:: torchani.data.BatchedANIDataset
Gao, Xiang's avatar
AEV Cache, part2 (#89)  
Gao, Xiang committed
18 
.. autoclass:: torchani.data.AEVCacheLoader
Gao, Xiang's avatar
Add tutorials (#86)  
Gao, Xiang committed
19
20
21
22
23
24
25 
.. automodule:: torchani.data.cache_aev


Utilities
=========

.. automodule:: torchani.utils
Gao, Xiang's avatar
Make Container able to handle aev cache, add benchmark for aev cache (#90)  
Gao, Xiang committed
26
27 
.. autofunction:: torchani.utils.pad
.. autofunction:: torchani.utils.pad_coordinates
Gao, Xiang's avatar
Add tutorials (#86)  
Gao, Xiang committed
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61 
.. autofunction:: torchani.utils.present_species
.. autofunction:: torchani.utils.strip_redundant_padding


NeuroChem
=========

.. automodule:: torchani.neurochem
.. autoclass:: torchani.neurochem.Constants
    :members:
.. autofunction:: torchani.neurochem.load_sae
.. autofunction:: torchani.neurochem.load_atomic_network
.. autofunction:: torchani.neurochem.load_model
.. autofunction:: torchani.neurochem.load_model_ensemble
.. autoclass:: torchani.neurochem.Builtins
.. autoclass:: torchani.neurochem.Trainer
    :members:
.. automodule:: torchani.neurochem.trainer


Ignite Helpers
==============

.. automodule:: torchani.ignite
.. autoclass:: torchani.ignite.Container
    :members:
.. autoclass:: torchani.ignite.DictLoss
.. autoclass:: torchani.ignite.PerAtomDictLoss
.. autoclass:: torchani.ignite.TransformedLoss
.. autofunction:: torchani.ignite.MSELoss
.. autoclass:: torchani.ignite.DictMetric
.. autofunction:: torchani.ignite.RMSEMetric
.. autoclass:: torchani.ignite.MaxAbsoluteError
.. autofunction:: torchani.ignite.MAEMetric