From 87737e9520125870b29bd6e5954290df6c77c1a6 Mon Sep 17 00:00:00 2001
From: Zofia Galla <zosiagalla@gmail.com>
Date: Sat, 15 May 2021 16:20:20 +0200
Subject: [PATCH] Add learning_rate parameter

---
 Jenkinsfile_training     | 5 +++--
 ium_zadanie6_training.py | 3 +--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Jenkinsfile_training b/Jenkinsfile_training
index cd90468..afcc20d 100644
--- a/Jenkinsfile_training
+++ b/Jenkinsfile_training
@@ -16,8 +16,9 @@ pipeline {
 		stage('Sh script') {
 			steps {
                 sh 'chmod +x run_training.sh'
-				sh './run_training.sh'
-                
+				withEnv(["LEARNING_RATE=${params.LEARNING_RATE}"]) {
+				sh './run_training.sh LEARNING_RATE'
+                }
 			
 			}
 		}
diff --git a/ium_zadanie6_training.py b/ium_zadanie6_training.py
index 5a6286e..39092a0 100644
--- a/ium_zadanie6_training.py
+++ b/ium_zadanie6_training.py
@@ -13,8 +13,7 @@ x_train = movies_train.copy()
 y_train = x_train.pop('rottentomatoes_audience_score')
 x_train.pop('Unnamed: 0')
 
-learning_rate = 0.001 #sys.argv[1]
-
+learning_rate = float(sys.argv[1])
 model = Sequential()
 model.add(layers.Input(shape=(22,)))
 model.add(layers.Dense(64))