Skip to content

Commit

Permalink
reinforce
Browse files Browse the repository at this point in the history
  • Loading branch information
albertbou92 committed Apr 23, 2024
1 parent 802251c commit 5eed34e
Show file tree
Hide file tree
Showing 5 changed files with 407 additions and 1 deletion.
36 changes: 36 additions & 0 deletions scripts/reinforce/config_denovo.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
# Logging configuration
experiment_name: acegen
agent_name: reinvereinforcnt
log_dir: results
logger_backend: null # csv, wandb, tensorboard, or null
seed: 101

# Environment configuration
num_envs: 128 # Number of smiles to generate in parallel
total_smiles: 10_000

# Scoring function
molscore: MolOpt
molscore_include: ["Albuterol_similarity"]
custom_task: null # Requires molscore to be set to null

# Fix the beginning of the generated molecules
prompt: null # e.g. c1ccccc

# Architecture
model: gru # gru, lstm, or gpt2
# The default prior varies for each model. Refer to the README file in the root directory for more information.
# The default vocabulary varies for each prior. Refer to the README file in the root directory for more information.

# Optimizer configuration
lr: 0.0001
eps: 1.0e-08
weight_decay: 0.0

# Reinforce configuration
sigma: 120

# Data replay configuration
replay_buffer_size: 100
experience_replay: True
replay_batch_size: 10
39 changes: 39 additions & 0 deletions scripts/reinforce/config_fragment.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
# Logging configuration
experiment_name: acegen
agent_name: reinvent
log_dir: results
logger_backend: null # csv, wandb, tensorboard, or null
seed: 101

# Environment configuration
num_envs: 128 # Number of smiles to generate in parallel
total_smiles: 10_000

# Scoring function
molscore: MolOpt
molscore_include: ["Celecoxxib_rediscovery"]
custom_task: null # Requires molscore to be set to null

# Promptsmiles configuration
promptsmiles: c1(C)ccc(*)cc1.NS(=O)(=O)(*)
promptsmiles_optimize: True
promptsmiles_shuffle: True
promptsmiles_multi: False

# Architecture
model: gru # gru, lstm, or gpt2
# The default prior varies for each model. Refer to the README file in the root directory for more information.
# The default vocabulary varies for each prior. Refer to the README file in the root directory for more information.

# Optimizer configuration
lr: 0.0001
eps: 1.0e-08
weight_decay: 0.0

# Reinforce configuration
sigma: 120

# Data replay configuration
replay_buffer_size: 100
experience_replay: True
replay_batch_size: 10
32 changes: 32 additions & 0 deletions scripts/reinforce/config_scaffold.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
# Logging configuration
experiment_name: acegen
agent_name: reinvent
log_dir: results
logger_backend: null # csv, wandb, tensorboard, or null
seed: 101

# Environment configuration
num_envs: 128 # Number of smiles to generate in parallel
total_smiles: 10_000

# Scoring function
molscore: LibINVENT_Exp1
molscore_include: ["DRD2_SelRF_SubFilt_DF"]
custom_task: null # Requires molscore to be set to null

# Promptsmiles configuration
promptsmiles: N1(*)CCN(CC1)CCCCN(*)
promptsmiles_optimize: True
promptsmiles_shuffle: True
promptsmiles_multi: False

# Architecture
model: gru # gru, lstm, or gpt2
# The default prior varies for each model. Refer to the README file in the root directory for more information.
# The default vocabulary varies for each prior. Refer to the README file in the root directory for more information.

# Optimizer configuration
lr: 0.0001
eps: 1.0e-08
weight_decay: 0.0

Loading

0 comments on commit 5eed34e

Please sign in to comment.