Skip to content

Commit 06ccc97

Browse files
committed
Updated the default parameters to make the problem more realistic
1 parent 9354df2 commit 06ccc97

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

Pilot2/P2B1/p2b1_baseline_keras2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def run(GP):
8080

8181
# set the seed
8282
if GP['seed']:
83-
np.random.seed(7)
83+
np.random.seed(GP['seed'])
8484
else:
8585
np.random.seed(np.random.randint(10000))
8686

@@ -279,7 +279,7 @@ def step_decay(epoch):
279279
#### Train the Model
280280
if GP['train_bool']:
281281
ct = hf.Candle_Molecular_Train(molecular_model, molecular_encoder, data_files, mb_epochs, callbacks,
282-
batch_size=32, nbr_type=GP['nbr_type'], save_path=GP['save_path'],
282+
batch_size=batch_size, nbr_type=GP['nbr_type'], save_path=GP['save_path'],
283283
len_molecular_hidden_layers=len_molecular_hidden_layers,
284284
molecular_nbrs=molecular_nbrs,
285285
conv_bool=conv_bool,

Pilot2/P2B1/p2b1_default_model.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[Global_Params]
22
num_hidden=[512,32]
3-
batch_size=32
3+
batch_size=64
44
learning_rate=0.01
55
epochs=2
66
cool=False
@@ -11,9 +11,9 @@ loss='custom'
1111
activation='relu'
1212
molecular_nonlinearity='elu'
1313
molecular_num_hidden=[256, 128, 64, 32, 16, 8]
14-
molecular_nbrs = 10
14+
molecular_nbrs = 200
1515
base_memo='p2b1'
1616
drop_prob = 0.5
1717
data_set='3k_Ordered'
18-
sampling_density = 0.05
18+
sampling_density = 0.15
1919
save_path='.'

0 commit comments

Comments
 (0)