From d566b3869a858cf23ad8ccfe95d90fc3832ad440 Mon Sep 17 00:00:00 2001
From: Dawid <dawid.wietrzych@gmail.com>
Date: Mon, 10 May 2021 19:36:19 +0200
Subject: [PATCH] added Jenkinsfile_train and model saving

---
 Jenkisfile_train | 35 +++++++++++++++++++++++++++++++++++
 tensor.py        |  4 +++-
 2 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 Jenkisfile_train

diff --git a/Jenkisfile_train b/Jenkisfile_train
new file mode 100644
index 0000000..fcab4a9
--- /dev/null
+++ b/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'
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/tensor.py b/tensor.py
index 4a4dd45..72a6738 100644
--- a/tensor.py
+++ b/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()
\ No newline at end of file
+sys.stdout.close()
+model.save('vaccines_model')
\ No newline at end of file