aboutsummaryrefslogtreecommitdiff
path: root/app/prod/config.json
blob: 5a09acc1a786e9fab75845199ee2671da8a7771a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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"
}