work on lab7
This commit is contained in:
parent
643b26ff72
commit
8e23b9517b
@ -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
|
# Przykładowo zwracamy coś w charakterze wyników, żeby było widoczne w plikach zapisanych przez obserwator
|
||||||
hist = pd.DataFrame(history.history)
|
hist = pd.DataFrame(history.history)
|
||||||
hist['epoch'] = history.epoch
|
hist['epoch'] = history.epoch
|
||||||
_run.info["train_results"] = str(hist.tail())
|
_run.log_scalar('final.training.loss', hist['loss'].iloc[-1])
|
||||||
return hist.tail()
|
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):
|
||||||
print(train())
|
train()
|
Loading…
Reference in New Issue
Block a user