IUM_06

Jenkinsfile

@@ -1,54 +1,39 @@
 pipeline {
-  agent any
+  agent { 
+    dockerfile true 
+  }
   
   parameters {
-    string (
-      defaultValue: 'vskyper',
-      description: 'Kaggle username',
-      name: 'KAGGLE_USERNAME',
-      trim: false
-    )
-    password (
-      defaultValue: '',
-      description: 'Kaggle API key',
-      name: 'KAGGLE_KEY',
+    buildSelector(
+      defaultSelector: lastSuccessful(),
+      description: 'Which build to use for copying artifacts',
+      name: 'BUILD_SELECTOR'
     )
   }
 
   stages {
     stage('Clone Repository') {
       steps {
-        git branch: 'main', url: 'https://git.wmi.amu.edu.pl/s464913/ium_464913.git'
+        git branch: 'training', url: 'https://git.wmi.amu.edu.pl/s464913/ium_464913.git'
       }
     }
 
-    stage('Download dataset') {
+    stage('Copy Artifacts') {
       steps {
-        withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
-          sh 'pip install kaggle'
-          sh 'kaggle datasets download -d mlg-ulb/creditcardfraud'
-          sh 'unzip -o creditcardfraud.zip'
-          sh 'rm creditcardfraud.zip'
-        }
-      }
-    }
-
-    stage('Run create-dataset script') {
-        agent {
-          dockerfile {
-            reuseNode true
+        copyArtifacts filter: 'data/*', projectName: 'z-s464913-create-dataset', selector: buildParameter('BUILD_SELECTOR')
       }
     }
 
+    stage('Run train_model script') {
       steps {
-        sh 'chmod +x create-dataset.py'
-        sh 'python3 ./create-dataset.py'
+        sh 'chmod +x train_model.py'
+        sh 'python3 ./train_model.py'
       }
     }
 
     stage('Archive Artifacts') {
       steps {
-        archiveArtifacts artifacts: 'data/*', onlyIfSuccessful: true
+        archiveArtifacts artifacts: 'model/*', onlyIfSuccessful: true
       }
     }
   }