diff options
| author | Navan Chauhan <navanchauhan@gmail.com> | 2020-09-24 16:11:25 +0530 |
|---|---|---|
| committer | Navan Chauhan <navanchauhan@gmail.com> | 2020-09-24 16:11:25 +0530 |
| commit | 7e9940d6c7fcca0109e90b3d68ecd68676635119 (patch) | |
| tree | 06d1ba55dd5ac24f86021ab59994114e6c4ac7e9 /app/prod/config.json | |
| parent | ebd31b9f60ea93391e3eb7d58c1a6fe0b813c6a9 (diff) | |
replaced dataset with one trained on 1.5 mil compounds
Diffstat (limited to 'app/prod/config.json')
| -rw-r--r-- | app/prod/config.json | 16 |
1 files changed, 8 insertions, 8 deletions
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 |