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'