added Jenkinsfile_train and model saving

Jenkisfile_train

@@ -0,0 +1,35 @@
+pipeline {
+    agent {dockerfile true}
+	parameters {
+		buildSelector(
+            defaultSelector: lastSuccessful(),
+            description: 'Which build to use for copying artifacts',
+            name: 'BUILD_SELECTOR'
+        )
+	}
+    stages {
+		stage('copyArtifacts') {
+            steps {
+                copyArtifacts fingerprintArtifacts: true, projectName: 's434804-create-dataset', selector: buildParameter('BUILD_SELECTOR')
+            }
+        }
+        stage('training') {
+			steps {
+                sh 'chmod +x tensor.py'
+                sh 'python tensor.py'
+			}
+		}
+		stage('archiveArtifacts') {
+			steps{
+				archiveArtifacts 'vaccines_model/saved_model.pb'
+			}
+		}
+        stage('sendMail') {
+            steps{
+                emailext body: currentBuild.result ?: 'TRAINING SUCCESS',
+                subject: 's434804',
+                to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
+            }
+        }
+    }
+}

tensor.py

@@ -2,6 +2,7 @@ import numpy as np
 import pandas as pd
 import tensorflow as tf 
 import sys
+import wget
 from tensorflow import keras 
 from sklearn.metrics import r2_score, mean_squared_error
 from math import sqrt
@@ -39,4 +40,5 @@ model.fit(X_train, y_train, epochs=50, validation_split = 0.3)
 
 prediction = model.predict(X_test)
 print(prediction)
-sys.stdout.close()
+sys.stdout.close()
+model.save('vaccines_model')