From 40d06776f5085ab16141118d6bd428a2a9e57fa2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Krzysztof=20Raczy=C5=84ski?= <krzysiek.r@onet.pl>
Date: Mon, 13 May 2024 20:10:55 +0200
Subject: [PATCH] fix model.py

---
 Jenkinsfile | 19 +++++++------------
 model.py    |  2 +-
 2 files changed, 8 insertions(+), 13 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index b448897..393f1ae 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,21 +1,21 @@
 pipeline {
-    agent { 
-        dockerfile true 
+    agent {
+        dockerfile true
     }
 
     parameters {
         string(
 			defaultValue: '10000',
-			name: 'CUTOFF',  
+			name: 'CUTOFF',
 			description: 'Liczba wierszy do obcięcia ze zbioru danych')
 	    string(
 			defaultValue: '',
-			name: 'KAGGLE_USERNAME',  
+			name: 'KAGGLE_USERNAME',
 			description: 'Kaggle username')
         password(
 			defaultValue: '',
-			name: 'KAGGLE_KEY',  
-			description: 'Kaggle API key') 
+			name: 'KAGGLE_KEY',
+			description: 'Kaggle API key')
     }
 
     stages {
@@ -28,7 +28,7 @@ pipeline {
         stage('Download') {
             steps {
                 withEnv([
-                    "KAGGLE_USERNAME=${env.KAGGLE_USERNAME}", 
+                    "KAGGLE_USERNAME=${env.KAGGLE_USERNAME}",
                     "KAGGLE_KEY=${env.KAGGLE_KEY}"
                 ]) {
                     sh "bash ./create-dataset.sh ${params.CUTOFF}"
@@ -44,10 +44,5 @@ pipeline {
                 archiveArtifacts artifacts: 'car_prices_predict_model.h5, predicted_selling_prices.csv', onlyIfSuccessful: true
             }
         }
-        stage('Archive Results') {
-            steps {
-                archiveArtifacts artifacts: 'data/*', onlyIfSuccessful: true
-            }
-        }
     }
 }
\ No newline at end of file
diff --git a/model.py b/model.py
index b2ffddc..3a8fd10 100644
--- a/model.py
+++ b/model.py
@@ -30,4 +30,4 @@ model.compile(optimizer='adam', loss='mean_squared_error')
 
 model.fit(X_train, y_train, epochs=20, batch_size=32)
 
-model.save('car_prices_predict_model.h5')
+model.save('./car_prices_predict_model.h5')