finał 7?

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

View File

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

View File

@ -67,10 +67,16 @@ def prepare_model(epoch_param, batch_param, _run):
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
def my_main(epoch_param, batch_param):
print(prepare_model())
r = ex.run()
ex.add_artifact("saved_model/saved_model.pb")
ex.add_artifact("wine_model.h5")