jenkins train model

Dockerfile

@@ -8,4 +8,4 @@ RUN pip3 install kaggle pandas scikit-learn
 
 WORKDIR /app
 
-COPY IUM_2.py ./
+COPY IUM_2.py model.py prediction.py ./

Jenkinsfile

@@ -35,5 +35,16 @@ pipeline {
              }
          }
          }
+
+         stage('Train and Predict') {
+            steps {
+               def customImage = docker.build("custom-image")
+                    customImage.inside {
+                        sh 'python3 ./model.py'
+                        sh 'python3 ./prediction.py'
+                        archiveArtifacts artifacts: 'model.pth, predictions.txt', onlyIfSuccessful: true
+                    }
+            }
+        }
      }
 }