From 33491f6458aa68ffb4c40ce1f9cb0d9156cf404c Mon Sep 17 00:00:00 2001
From: Alagris <alagris12358@gmail.com>
Date: Mon, 10 May 2021 12:37:12 +0200
Subject: [PATCH] no mongo

---
 evaluation.Jenkinsfile  | 8 ++++++--
 train-model.Jenkinsfile | 8 ++++++--
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/evaluation.Jenkinsfile b/evaluation.Jenkinsfile
index b470c9f..48d4d00 100644
--- a/evaluation.Jenkinsfile
+++ b/evaluation.Jenkinsfile
@@ -2,13 +2,17 @@ pipeline {
     agent { 
         dockerfile true 
     }
-
+    parameters {
+        string(name: 'batch_size', defaultValue: '512', description: 'size of batch')
+        string(name: 'learning_rate', defaultValue: '0.01', description: 'Learning rate')
+        string(name: 'epochs', defaultValue: '14', description: 'epochs')
+    }
     stages {
         stage('Build') {
             steps {
                 git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git'
                 copyArtifacts fingerprintArtifacts: true, projectName: 's434749-training', selector: lastSuccessful()
-                sh 'python3 train_model.py with "mode=eval"'
+                sh "python3 train_model.py with 'mode=eval' 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'"
                 script{
                     def results = readFile "${env.WORKSPACE}/results.txt"
                 }
diff --git a/train-model.Jenkinsfile b/train-model.Jenkinsfile
index 4f129d0..4507346 100644
--- a/train-model.Jenkinsfile
+++ b/train-model.Jenkinsfile
@@ -2,13 +2,17 @@ pipeline {
     agent { 
         dockerfile true 
     }
-
+    parameters {
+        string(name: 'batch_size', defaultValue: '512', description: 'size of batch')
+        string(name: 'learning_rate', defaultValue: '0.01', description: 'Learning rate')
+        string(name: 'epochs', defaultValue: '14', description: 'epochs')
+    }
     stages {
         stage('Build') {
             steps {
                 git 'https://git.wmi.amu.edu.pl/s434749/ium_434749.git'
                 copyArtifacts fingerprintArtifacts: true, projectName: 's434749-create-dataset', selector: lastSuccessful()
-                sh 'python3 train_model.py'
+                sh "python3 train_model.py with 'batch_size=${params.batch_size}' 'learning_rate=${params.learning_rate}' 'epochs=${params.epochs}'"
             }
 
             post {