just loss
This commit is contained in:
parent
407d6187af
commit
09cbfad0ef
@ -46,7 +46,7 @@ def prepare_train_model(learning_rate, _run):
|
||||
|
||||
model.save('grid-stability-dense.h5')
|
||||
|
||||
_run['history'] = str(history.history[:, -1])
|
||||
_run['history'] = str(history.history['loss'][1])
|
||||
|
||||
|
||||
@ex.main
|
||||
|
Loading…
Reference in New Issue
Block a user