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

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

Browse Source
This commit is contained in:
Kacper Dudzic 2022-05-06 22:30:53 +02:00
parent c3729ed7cd
commit 157c74ef21
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lab7/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):