diff options
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 |