Update 'simple_regression_lab7.py'
This commit is contained in:
parent
fcd9be7cc7
commit
732499e4bc
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user