s434704/ium_434704
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix casting

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-05-14 04:12:42 +02:00
parent f1083d4e7a
commit e67ad3779c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
training.py
View File

@ -44,8 +44,8 @@ else:
optimizer=tf.keras.optimizers.Adam(0.001),
metrics=[tf.keras.metrics.RootMeanSquaredError()])
verbose = 0 if len(verbose) == 0 else verbose[0]
epochs = 100 if len(epochs) == 0 else epochs[0]
verbose = 0 if len(verbose) == 0 else int(verbose[0])
epochs = 100 if len(epochs) == 0 else int(epochs[0])
model.fit(train_X, train_Y, verbose=verbose, epochs=epochs)
model.save('linear_regression.h5')