kdudzic/ium_s449288
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'simple_regression_lab7.py'

Browse Source
This commit is contained in:
Kacper Dudzic 2022-05-06 19:57:31 +02:00
parent fcd9be7cc7
commit 732499e4bc
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
simple_regression_lab7.py
View File

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