Skip to content

Commit a97661f

Browse files
committed
update graphga oracle doc
1 parent 52a2797 commit a97661f

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

docs/source/api/generator.rst

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -58,17 +58,13 @@ Modeling Molecules as Graphs
5858
:undoc-members:
5959
:show-inheritance:
6060

61-
Modeling Molecules as Graphs with Heuristic Methods
62-
---------------------------------------------------
63-
6461
.. rubric:: Graph Genetic Algorithm for Un/Multi-conditional Molecular Generation
6562
.. autoclass:: torch_molecule.generator.graph_ga.modeling_graph_ga.GraphGAMolecularGenerator
6663
:exclude-members: fitting_epoch, fitting_loss, save_to_hf, load_from_hf
6764
:members: fit, generate
6865
:undoc-members:
6966
:show-inheritance:
70-
71-
.. rubric:: Default Oracles in GraphGA
67+
7268
.. automodule:: torch_molecule.generator.graph_ga.oracle
7369
:members:
7470
:undoc-members:

torch_molecule/generator/graph_ga/oracle.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77

88
class Oracle:
9-
"""Oracle class for scoring molecules.
9+
"""The default Oracle class for scoring molecules in GraphGA.
1010
1111
This class wraps predictive models (like RandomForestRegressor) to score molecules
1212
based on their properties. It handles conversion of SMILES to fingerprints.

0 commit comments

Comments
 (0)