work on lab7
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 2022-05-06 19:23:29 +02:00
parent 643b26ff72
commit 8e23b9517b

View File

@ -81,9 +81,10 @@ def train(units, learning_rate, _run):
# Przykładowo zwracamy coś w charakterze wyników, żeby było widoczne w plikach zapisanych przez obserwator
hist = pd.DataFrame(history.history)
hist['epoch'] = history.epoch
_run.info["train_results"] = str(hist.tail())
return hist.tail()
_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):
print(train())
train()