From caa98f339fb11b3a7b12f3fd5df1422daea27fb2 Mon Sep 17 00:00:00 2001 From: futianfan Date: Sun, 21 May 2023 23:51:07 -0400 Subject: [PATCH] smiles_aug_mem, smiles_bar in run.py --- run.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/run.py b/run.py index 4d21cd9..ad42016 100644 --- a/run.py +++ b/run.py @@ -63,6 +63,10 @@ def main(): from main.graph_mcts.run import Graph_MCTS_Optimizer as Optimizer elif args.method == 'smiles_ahc': from main.smiles_ahc.run import AHC_Optimizer as Optimizer + elif args.method == 'smiles_aug_mem': + from main.smiles_aug_mem.run import AugmentedMemory_Optimizer as Optimizer + elif args.method == 'smiles_bar': + from main.smiles_bar.run import BAR_Optimizer as Optimizer elif args.method == "smiles_lstm_hc": from main.smiles_lstm_hc.run import SMILES_LSTM_HC_Optimizer as Optimizer elif args.method == 'selfies_lstm_hc':