update readme

Author: liugangcode

docs/source/api/encoder.rst

@@ -1,42 +1,45 @@
 Molecular Encoder Models
 ====================================================
 
-This section documents the encoder(representation learning) models in `torch_molecule.encoder`.
-
 .. contents::
    :local:
    :depth: 1
 
 .. rubric:: MoAma for Molecular Representation Learning
 .. automodule:: torch_molecule.encoder.moama
    :members: 
+   :exclude-members: fitting_epoch, fitting_loss
    :undoc-members:
    :show-inheritance:
 
 .. rubric:: Supervised/Pseudolabeled Pretraining for Molecules
 
 .. automodule:: torch_molecule.encoder.supervised
    :members:
+   :exclude-members: fitting_epoch, fitting_loss
    :undoc-members:
    :show-inheritance:
 
 .. rubric:: Attribute Masking for Molecular Representation Learning
 
 .. automodule:: torch_molecule.encoder.attrmask
    :members:
+   :exclude-members: fitting_epoch, fitting_loss
    :undoc-members:
    :show-inheritance:
 
 .. rubric:: Context Prediction for Molecular Representation Learning
 
 .. automodule:: torch_molecule.encoder.contextpred
    :members:
+   :exclude-members: fitting_epoch, fitting_loss
    :undoc-members:
    :show-inheritance:
 
 .. rubric:: Edge Prediction for Molecular Representation Learning
 
 .. automodule:: torch_molecule.encoder.edgepred
    :members:
+   :exclude-members: fitting_epoch, fitting_loss
    :undoc-members:
    :show-inheritance:

docs/source/api/generator.rst

@@ -1,17 +1,15 @@
 Molecular Generation Models
 =============================
 
-This section documents the generation models in `torch_molecule.generator`.
-
 .. contents::
    :local:
    :depth: 1
-   
+
 .. rubric:: GraphDiT for Un/Multi-conditional Molecular Generation
 
 .. automodule:: torch_molecule.generator.graph_dit
    :members:
-   :exclude-members: fitting_epoch, fitting_loss
+   :exclude-members: fitting_epoch, fitting_loss, model_class, dataset_info, model_name
    :undoc-members:
    :show-inheritance:
 
@@ -20,14 +18,14 @@ This section documents the generation models in `torch_molecule.generator`.
 
 .. automodule:: torch_molecule.generator.digress
    :members:
-   :exclude-members: fitting_epoch, fitting_loss
+   :exclude-members: fitting_epoch, fitting_loss, model_class, dataset_info, model_name
    :undoc-members:
    :show-inheritance:
 
 .. rubric:: Graph Genetic Algorithm for Un/Multi-conditional Molecular Generation
 
 .. automodule:: torch_molecule.generator.graphga
    :members:
-   :exclude-members: fitting_epoch, fitting_loss
+   :exclude-members: fitting_epoch, fitting_loss, push_to_huggingface, load_from_huggingface
    :undoc-members:
    :show-inheritance:

docs/source/api/predictor.rst

@@ -1,14 +1,10 @@
 Molecular Property Prediction Models
 =====================================
 
-This section documents the predictor models in `torch_molecule.predictor`.
-
 .. contents::
    :local:
    :depth: 1
 
-.. (Add predictors here as needed. You might write summaries and link submodules.)
-
 .. rubric:: Graph Neural Networks
 .. .. autoclass:: torch_molecule.predictor.gnn.modeling_gnn.GNNMolecularPredictor
 ..    :members: fit, autofit, predict

docs/source/index.rst

@@ -14,6 +14,7 @@ Welcome to the torch-molecule documentation. torch-molecule is an actively devel
    :maxdepth: 2
    :caption: Getting Started:
 
+   index
    install
    example