InternLM/InternBootcamp
1
0
Fork 0
mirror of https://github.com/InternLM/InternBootcamp.git synced 2026-04-19 12:58:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(bootcamp): simplify InChI and SMILES generation

Browse source 
- 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
Download patch file Download diff file Expand all files Collapse all files

7
internbootcamp/bootcamp/ChemStructure2Property/InChI2logPBootCamp.py
View file

@ -20,11 +20,10 @@ class InChI2logPbootcamp(Basebootcamp):
生成一组数字和目标值。
"""
self.InChIGenerator = InChIGenerator(max_atoms=self.max_atoms, min_atoms=self.min_atoms, elements=None, seed=None)
inchis = self.InChIGenerator.generate_n_valid_inchi(10)
inchis = self.InChIGenerator.generate_n_valid_inchi(1)
# print(inchis)
n = random.randint(0, 9)
# print(n)
return inchis[n]
return inchis[0]
def prompt_func(self, InChI) -> str:
@ -76,7 +75,7 @@ class InChI2logPbootcamp(Basebootcamp):
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
if __name__ == "__main__":