Update 'simple_regression_lab7.py'

This commit is contained in:
Kacper Dudzic 2022-05-06 19:57:31 +02:00
parent fcd9be7cc7
commit 732499e4bc

View File

@ -80,8 +80,6 @@ def train(epochs, units, learning_rate, _run):
hist = pd.DataFrame(history.history)
hist['epoch'] = history.epoch
_run.log_scalar('final.training.loss', hist['loss'].iloc[-1])
print(hist['loss'].iloc[-1])
print(type(hist['loss'].iloc[-1]))
@ex.automain
def main(units, learning_rate):