update docs

Author: liugangcode

docs/source/api/base.rst

@@ -8,7 +8,7 @@ This section documents base classes in `torch_molecule.base`.
    :undoc-members:
    :show-inheritance:
 
-.. automodule:: torch_molecule.base.encoder
+.. automodule:: torch_molecule.base.predictor
    :members:
    :undoc-members:
    :show-inheritance:
@@ -18,7 +18,7 @@ This section documents base classes in `torch_molecule.base`.
    :undoc-members:
    :show-inheritance:
 
-.. automodule:: torch_molecule.base.predictor
+.. automodule:: torch_molecule.base.encoder
    :members:
    :undoc-members:
-   :show-inheritance:
+   :show-inheritance:

docs/source/index.rst

@@ -30,9 +30,8 @@ Welcome to the torch-molecule documentation. torch-molecule is an actively devel
    :caption: API for utilities:
 
    api/utils
-   api/gnn
-   api/generator
-   api/predictor
+   api/nn
+   api/base
 
 .. toctree::
    :maxdepth: 2