diff --git a/Jenkinsfile b/Jenkinsfile
index 1dbd804..0924d7f 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -27,6 +27,15 @@ pipeline {
                copyArtifacts fingerprintArtifacts: true, projectName: 's434695-create-dataset', selector: buildParameter('WHICH_BUILD')
             }
         }
+        stage('train')
+        {
+            steps
+                {
+                    catchError {
+                        sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
+                    }
+                }
+        }
       stage('archiveArtifacts') {
 			steps{
 				archiveArtifacts 'vgsales_model.h5'