From 1189623cdbc0a00babf69d2b95fd7ab841437611 Mon Sep 17 00:00:00 2001
From: Dominik Strzako <sylvaripl@gmail.com>
Date: Sat, 15 May 2021 20:42:00 +0200
Subject: [PATCH] =?UTF-8?q?fina=C5=82=207=3F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Jenkinsfile_train      | 1 +
 Zadanie_06_training.py | 8 +++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index cae67f7..76a6275 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -39,6 +39,7 @@ pipeline {
 		stage('Archive artifacts') {
 			steps{
 				archiveArtifacts 'wine_model.h5'
+                archiveArtifacts 'my_runs/**'
 			}
 		}
 	}
diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py
index 6b908a8..1a36a82 100644
--- a/Zadanie_06_training.py
+++ b/Zadanie_06_training.py
@@ -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")
\ No newline at end of file
+ex.add_artifact("wine_model.h5")
\ No newline at end of file