str
This commit is contained in:
parent
b60408e2e8
commit
c815688698
@ -36,14 +36,14 @@ def prepare_train_model(learning_rate, _run):
|
||||
|
||||
model.compile(
|
||||
loss=tensorflow.keras.losses.BinaryCrossentropy(),
|
||||
optimizer=tensorflow.keras.optimizers.Adam(lr=learning_rate),
|
||||
optimizer=tensorflow.keras.optimizers.Adam(learning_rate=learning_rate),
|
||||
metrics=[tensorflow.keras.metrics.BinaryAccuracy()])
|
||||
|
||||
history = model.fit(X_train, Y_train, epochs=2, validation_data=(X_valid, Y_valid))
|
||||
|
||||
model.save('grid-stability-dense.h5')
|
||||
|
||||
_run['history'] = history
|
||||
_run['history'] = str(history)
|
||||
|
||||
@ex.main
|
||||
def my_main(learning_rate):
|
||||
|
Loading…
Reference in New Issue
Block a user