From 7e9940d6c7fcca0109e90b3d68ecd68676635119 Mon Sep 17 00:00:00 2001 From: Navan Chauhan Date: Thu, 24 Sep 2020 16:11:25 +0530 Subject: replaced dataset with one trained on 1.5 mil compounds --- app/prod/config.json | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'app/prod/config.json') diff --git a/app/prod/config.json b/app/prod/config.json index 5a09acc..cc46468 100644 --- a/app/prod/config.json +++ b/app/prod/config.json @@ -1,9 +1,9 @@ { "exp_name": "LSTM_Chem", - "data_filename": "./datasets/all_smiles_clean.txt", + "data_filename": "./datasets/1500S_SMILES.txt", "data_length": 0, "units": 256, - "num_epochs": 22, + "num_epochs": 1, "optimizer": "adam", "seed": 71, "batch_size": 256, @@ -21,10 +21,10 @@ "finetune_batch_size": 1, "finetune_data_filename": "./datasets/protease_inhibitors_for_fine-tune.txt", "config_file": "app/prod/config.json", - "exp_dir": "experiments/alpha/LSTM_Chem", - "tensorboard_log_dir": "experiments/alpha/LSTM_Chem/logs/", - "checkpoint_dir": "experiments/alpha/LSTM_Chem/checkpoints/", - "train_smi_max_len": 128, + "exp_dir": "experiments/2020-09-24/LSTM_Chem", + "tensorboard_log_dir": "experiments/2020-09-24/LSTM_Chem/logs/", + "checkpoint_dir": "experiments/2020-09-24/LSTM_Chem/checkpoints/", + "train_smi_max_len": 1429, "model_arch_filename": "app/prod/model_arch.json", - "model_weight_filename": "app/prod/checkpoints/LSTM_Chem-22-0.24.hdf5" -} + "model_weight_filename": "app/prod/checkpoints/LSTM_Chem-01-0.03.hdf5" +} \ No newline at end of file -- cgit v1.2.3