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

add learn parameter
Some checks failed
s444417-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
s444417 2022-04-25 22:55:03 +02:00
parent cda767977a
commit 41d0a23830
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
runPythonScripts.sh
View File

@ -1,3 +1,3 @@
#!/bin/bash #!/bin/bash
python3 ./src/task1python.py python3 ./src/task1python.py
python3 ./src/trainScript.py python3 ./src/trainScript.py 6

5
src/trainScript.py
View File

@ -6,6 +6,9 @@ import numpy as np
import tensorflow as tf import tensorflow as tf
from tensorflow.keras import layers from tensorflow.keras import layers
#train params
numberOfEpoch = sys.argv[1]
cwd = os.path.abspath(os.path.dirname(sys.argv[0])) cwd = os.path.abspath(os.path.dirname(sys.argv[0]))
pathTrain = cwd + "/../Participants_Data_HPP/Train.csv" pathTrain = cwd + "/../Participants_Data_HPP/Train.csv"
@ -56,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=5, ) history = linear_model.fit(house_price_features, house_price_labels, epochs=numberOfEpoch, )
#callbacks=[cp_callback]) #callbacks=[cp_callback])
# print(history) # print(history)