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

save model
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:09:52 +02:00
parent ce6278c648
commit 10e3f9fad8
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@ -16,10 +16,10 @@ WORKDIR /app
COPY . . COPY . .
RUN ls -la
RUN chmod o+wrx ./startscript1.sh RUN chmod o+wrx ./startscript1.sh
RUN chmod o+wrx ./runPythonScripts.sh RUN chmod o+wrx ./runPythonScripts.sh
RUN ./startscript1.sh RUN ./startscript1.sh
RUN ls -la

5
src/trainScript.py
View File

@ -56,10 +56,13 @@ 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=10, ) history = linear_model.fit(house_price_features, house_price_labels, epochs=5, )
#callbacks=[cp_callback]) #callbacks=[cp_callback])
# print(history) # print(history)
# save model
linear_model.save('saved_model/my_model')
test_results = {} test_results = {}
test_results['linear_model'] = linear_model.evaluate( test_results['linear_model'] = linear_model.evaluate(
house_price_test_features, house_price_test_expected, verbose=0) house_price_test_features, house_price_test_expected, verbose=0)