refactor(bootcamp): simplify InChI and SMILES generation

- Reduce the number of generated InChIs and SMILES from 10 to 1
- Remove random selection, always return the first generated structure
- Comment out debug prints and unused code
This commit is contained in:
chenyongkang 2025-06-17 13:29:31 +08:00
parent f4913c6f02
commit 18f47e0a3a
3 changed files with 7 additions and 8 deletions

View file

@ -21,7 +21,7 @@ class SMILES2logPbootcamp(InChI2logPbootcamp):
生成一组数字和目标值
"""
self.SMILESGenerator = SMILESGenerator(min_len=self.min_len, max_len=self.max_len, seed=None)
return self.SMILESGenerator.generate_n_valid_smiles(10)[random.randint(0, 9)]
return self.SMILESGenerator.generate_n_valid_smiles(1)[0]
def prompt_func(self, SMILES) -> str:
@ -63,5 +63,5 @@ class SMILES2logPbootcamp(InChI2logPbootcamp):
return 0.0 # Error is too large, score is 0
else:
# Linear interpolation: score decreases linearly from 1 to 0 as error goes from 0 to max_relative_error
return 1.0
# return 1.0
return 1 - (relative_error / max_relative_error) * 0.5 ## For RL