From e80badd278f54e8cde64243aa56687f8e6e32cc2 Mon Sep 17 00:00:00 2001 From: Navan Chauhan Date: Sat, 1 Aug 2020 15:34:22 +0530 Subject: added alpha model --- app/prod/config.json | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'app/prod/config.json') diff --git a/app/prod/config.json b/app/prod/config.json index ccab45f..5a09acc 100644 --- a/app/prod/config.json +++ b/app/prod/config.json @@ -3,10 +3,10 @@ "data_filename": "./datasets/all_smiles_clean.txt", "data_length": 0, "units": 256, - "num_epochs": 42, + "num_epochs": 22, "optimizer": "adam", "seed": 71, - "batch_size": 512, + "batch_size": 256, "validation_split": 0.1, "verbose_training": true, "checkpoint_monitor": "val_loss", @@ -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/2020-07-13/LSTM_Chem", - "tensorboard_log_dir": "app/prod/logs/", - "checkpoint_dir": "app/prod/checkpoints/", + "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-42-0.23.hdf5" -} \ No newline at end of file + "model_weight_filename": "app/prod/checkpoints/LSTM_Chem-22-0.24.hdf5" +} -- cgit v1.2.3