diff options
| author | Navan Chauhan <navanchauhan@gmail.com> | 2020-07-31 21:23:03 +0530 |
|---|---|---|
| committer | Navan Chauhan <navanchauhan@gmail.com> | 2020-07-31 21:23:03 +0530 |
| commit | 61ce4e7b089d68395be2221f64d89040c0b14a34 (patch) | |
| tree | 6bd60c1fd4f8bcd1c503914c61272ed382fff1da /app/prod/config.json | |
| parent | 376f04d1df2692a8874de686372d18b9ab07950a (diff) | |
added AI model
Diffstat (limited to 'app/prod/config.json')
| -rw-r--r-- | app/prod/config.json | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/app/prod/config.json b/app/prod/config.json new file mode 100644 index 0000000..4b72262 --- /dev/null +++ b/app/prod/config.json @@ -0,0 +1,30 @@ +{ + "exp_name": "LSTM_Chem", + "data_filename": "./datasets/all_smiles_clean.txt", + "data_length": 0, + "units": 256, + "num_epochs": 42, + "optimizer": "adam", + "seed": 71, + "batch_size": 512, + "validation_split": 0.1, + "verbose_training": true, + "checkpoint_monitor": "val_loss", + "checkpoint_mode": "min", + "checkpoint_save_best_only": false, + "checkpoint_save_weights_only": true, + "checkpoint_verbose": 1, + "tensorboard_write_graph": true, + "sampling_temp": 0.75, + "smiles_max_length": 128, + "finetune_epochs": 12, + "finetune_batch_size": 1, + "finetune_data_filename": "./datasets/protease_inhibitors_for_fine-tune.txt", + "config_file": "experiments/base_experiment/LSTM_Chem/config.json", + "exp_dir": "experiments/2020-07-13/LSTM_Chem", + "tensorboard_log_dir": "experiments/2020-07-13/LSTM_Chem/logs/", + "checkpoint_dir": "experiments/2020-07-13/LSTM_Chem/checkpoints/", + "train_smi_max_len": 128, + "model_arch_filename": "experiments/2020-07-13/LSTM_Chem/model_arch.json", + "model_weight_filename": "experiments/2020-07-13/LSTM_Chem/checkpoints/LSTM_Chem-42-0.23.hdf5" +}
\ No newline at end of file |