aboutsummaryrefslogtreecommitdiff
path: root/app/prod/config.json
diff options
context:
space:
mode:
Diffstat (limited to 'app/prod/config.json')
-rw-r--r--app/prod/config.json30
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..5a09acc
--- /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": 22,
+ "optimizer": "adam",
+ "seed": 71,
+ "batch_size": 256,
+ "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": "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,
+ "model_arch_filename": "app/prod/model_arch.json",
+ "model_weight_filename": "app/prod/checkpoints/LSTM_Chem-22-0.24.hdf5"
+}