just loss

This commit is contained in:
jakubknczny 2021-05-16 16:06:07 +02:00
parent 407d6187af
commit 09cbfad0ef

View File

@ -46,7 +46,7 @@ def prepare_train_model(learning_rate, _run):
model.save('grid-stability-dense.h5') model.save('grid-stability-dense.h5')
_run['history'] = str(history.history[:, -1]) _run['history'] = str(history.history['loss'][1])
@ex.main @ex.main