Skip to content

Commit 9da1f51

Browse files
committed
Merge branch 'develop' of github.com:ECP-CANDLE/Benchmarks into develop
2 parents 35bd722 + ec53a59 commit 9da1f51

18 files changed

+610
-274
lines changed

Pilot1/Combo/combo_default_model.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
[Global_Params]
2+
model_name='combo'
23
cell_features=['expression']
34
drug_features=['descriptors']
45
dense=[1000, 1000, 1000]

Pilot1/NT3/nt3_baseline_keras2.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,6 @@ def run(gParameters):
8585
model = Sequential()
8686

8787
initial_epoch = 0
88-
best_metric_last = None
8988

9089
X_train, Y_train, X_test, Y_test = load_data(train_file, test_file, gParameters)
9190

@@ -167,14 +166,12 @@ def run(gParameters):
167166
model.add(Dense(gParameters["classes"]))
168167
model.add(Activation(gParameters["out_activation"]))
169168

170-
J = candle.restart(gParameters, model)
169+
ckpt = candle.CandleCkptKeras(gParameters, verbose=False)
170+
ckpt.set_model(model)
171+
J = ckpt.restart(model)
171172
if J is not None:
172173
initial_epoch = J["epoch"]
173-
best_metric_last = J["best_metric_last"]
174-
gParameters["ckpt_best_metric_last"] = best_metric_last
175-
print("initial_epoch: %i" % initial_epoch)
176-
177-
ckpt = candle.CandleCheckpointCallback(gParameters, verbose=False)
174+
print("restarting from ckpt: initial_epoch: %i" % initial_epoch)
178175

179176
# Reference case
180177
# model.add(Conv1D(filters=128, kernel_size=20, strides=1, padding='valid', input_shape=(P, 1)))

Pilot1/NT3/train.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
#!/bin/bash
2+
3+
# TRAIN SH
4+
# IMPROVE interface to NT3 model

0 commit comments

Comments
 (0)