train model

Jenkins/Jenkinsfile.create_dataset

@@ -31,7 +31,6 @@ node {
             withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
                 "KAGGLE_KEY=${params.KAGGLE_KEY}","CUTOFF=${params.CUTOFF}"]) {
                 sh "python3 Scripts/download_dataset.py '.' 'dataset.csv'"
-                sh "python3 Scripts/train_neural_network.py '.'"
             }
         }
         stage('Archive artifacts') {

Jenkins/Jenkinsfile.training

@@ -0,0 +1,21 @@
+node {
+    checkout scm
+    docker.image('s444452/ium:1.3').inside {
+        stage('Preparation') {
+            properties([
+                parameters([
+                string(
+                 defaultValue: ".",
+                 description: 'Arguments for model training: arg1,arg2,arg3',
+                 name: 'TRAIN_ARGS'
+                )
+                ])
+            ])
+        }
+        stage('Run script') {
+            withEnv(["TRAIN_ARGS=${params.TRAIN_ARGS}"]) {
+                sh "python3 Scripts/train_neural_network.py $TRAIN_ARGS"
+            }
+        }
+    }
+}