From 4664e19d2eceef4ad82f4bc37c5ca84efac241cc Mon Sep 17 00:00:00 2001
From: Kacper Dudzic <kacdud1@st.amu.edu.pl>
Date: Sun, 24 Apr 2022 18:03:25 +0200
Subject: [PATCH] Update 'lab6/Jenkinsfile'

---
 lab6/Jenkinsfile | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/lab6/Jenkinsfile b/lab6/Jenkinsfile
index c82b414..6a92f08 100644
--- a/lab6/Jenkinsfile
+++ b/lab6/Jenkinsfile
@@ -2,6 +2,13 @@ pipeline {
     agent {
         dockerfile true
         }
+    parameters {
+        string(
+			defaultValue: '100',
+			description: 'Example training parameter',
+			name: 'EPOCHS_NUM'
+		)
+    }
     stages {
         stage('Stage 1') {
             steps {
@@ -10,7 +17,7 @@ pipeline {
 					copyArtifacts filter: '*', projectName: 's449288-create-dataset'
 					echo 'Datasets copied'
 					echo 'Conducting simple regression model test'
-					sh 'python3 simple_regression.py'
+					sh 'python3 simple_regression.py $EPOCHS_NUM'
 					echo 'Model and predictions saved'
                     sh 'head lego_reg_results.csv'
 					sh 'ls -lh lego_reg_model'