-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathconfig.py
28 lines (28 loc) · 943 Bytes
/
config.py
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
class Config(object):
data_filename = './dataset_cleaned.smi'
exp_name = 'LSTM_Chem'
data_length = 0
units = 256
num_epochs = 42
optimizer = 'adam'
seed = 71
batch_size = 128
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 = 5
finetune_batch_size = 1
finetune_data_filename = ''
exp_dir = './code_exp/LSTM_Chem/'
tensorboard_log_dir = './code_exp/LSTM_Chem/logs/'
checkpoint_dir = './code_exp/LSTM_Chem/checkpoints/'
train_smi_max_len = 128
model_arch_filename = './code_exp/LSTM_Chem/model_arch.json'
model_weight_filename = './code_exp/LSTM_Chem/checkpoints/LSTM_Chem-baseline-model-full.hdf5'