Update 'lab7/simple_regression_lab7.py'
All checks were successful
s449288-evaluation/pipeline/head This commit looks good
s449288-training/pipeline/head This commit looks good

This commit is contained in:
Kacper Dudzic 2022-05-06 22:30:53 +02:00
parent c3729ed7cd
commit 157c74ef21

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):