From 05b83727ff6874302113f3ee5d1cb4e799b9fba4 Mon Sep 17 00:00:00 2001
From: Wojciech Lidwin <59863785+Halal37@users.noreply.github.com>
Date: Fri, 12 May 2023 04:51:20 +0200
Subject: [PATCH] Fix

---
 Jenkinsfile_train | 2 +-
 ium_sacred.py     | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index b1071da..96b2294 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -25,7 +25,7 @@ node {
      checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
  }
  stage('Dockerfile'){
-    def testImage = docker.image('s487197/ium:50')
+    def testImage = docker.image('s487197/ium:51')
 	testImage.inside{ 
          copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
 		 sh "python3 ium_sacred.py -epochs $EPOCHS -lr $LR -validation_split $VALIDATION_SPLIT"
diff --git a/ium_sacred.py b/ium_sacred.py
index 4c54c11..f7364e5 100644
--- a/ium_sacred.py
+++ b/ium_sacred.py
@@ -78,9 +78,9 @@ def my_main(epochs, lr, validation_split, _run):
         validation_split=validation_split)
     hist = pd.DataFrame(history.history)
     hist['epoch'] = history.epoch
-    for his in hist.iterrows():
-        _run.log_scalar('training.loss', his[1]['loss'])
-        _run.log_scalar('accuracy', his[1]['accuracy'])
+    #for his in hist.iterrows():
+    _run.log_scalar('training.loss', hist['loss'])
+    _run.log_scalar('accuracy', hist['accuracy'])
     model.save('baltimore_model')
     ex.add_artifact('baltimore_model')