s434788/ium_434788
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

finał 7?

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-15 20:42:00 +02:00
parent aea643dee6
commit 1189623cdb
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Jenkinsfile_train
View File

@ -39,6 +39,7 @@ pipeline {
stage('Archive artifacts') { stage('Archive artifacts') {
steps{ steps{
archiveArtifacts 'wine_model.h5' archiveArtifacts 'wine_model.h5'
archiveArtifacts 'my_runs/**'
} }
} }
} }

8
Zadanie_06_training.py
View File

@ -67,10 +67,16 @@ def prepare_model(epoch_param, batch_param, _run):
NeuralModel.save('wine_model.h5') NeuralModel.save('wine_model.h5')
#TO TYLKO NA POTRZEBY ZADANIA
_run.info["Accuracy dla zbioru test (tylko na potrzeby zadania)"] = str(NeuralModel.predict(scaled_x))
@ex.main @ex.main
def my_main(epoch_param, batch_param): def my_main(epoch_param, batch_param):
print(prepare_model()) print(prepare_model())
r = ex.run() r = ex.run()
ex.add_artifact("saved_model/saved_model.pb") ex.add_artifact("wine_model.h5")