add infograph

liugangcode · liugangcode · commit 0f335dbfcc86 · 2025-04-03T22:26:16.000-04:00
diff --git a/docs/source/api/encoder.rst b/docs/source/api/encoder.rst
@@ -58,6 +58,14 @@ Self-supervised Molecular Representation Learning
    :undoc-members:
    :show-inheritance:
 
+.. rubric:: InfoGraph for Molecular Representation Learning
+
+.. autoclass:: torch_molecule.encoder.infograph.modeling_infograph.InfoGraphMolecularEncoder
+   :members: fit, encode
+   :exclude-members: fitting_epoch, fitting_loss, model_name, model_class
+   :undoc-members:
+   :show-inheritance:
+
 Supervised Pretraining for Molecules
 ------------------------------------
 
