s444417/ium_444417
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

convert parameter to int
All checks were successful
s444417-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
s444417 2022-04-25 22:56:21 +02:00
parent 41d0a23830
commit 3bf9949b17
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/trainScript.py
View File

@ -59,7 +59,7 @@ linear_model.compile(loss = tf.losses.MeanSquaredError(),
optimizer = tf.optimizers.Adam(1)) optimizer = tf.optimizers.Adam(1))
# train model # train model
history = linear_model.fit(house_price_features, house_price_labels, epochs=numberOfEpoch, ) history = linear_model.fit(house_price_features, house_price_labels, epochs=int(numberOfEpoch), )
#callbacks=[cp_callback]) #callbacks=[cp_callback])
# print(history) # print(history)