From 2d0e53d1eb5c6e8d347f6b96b14cb18e5bdc2d43 Mon Sep 17 00:00:00 2001
From: s434695 <tomgrz3@st.amu.edu.pl>
Date: Sat, 15 May 2021 18:24:30 +0200
Subject: [PATCH] sadge3

---
 Jenkinsfile | 2 +-
 train.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index 2469915..197a084 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -29,7 +29,7 @@ pipeline {
         }
       stage('archiveArtifacts') {
 			steps{
-				archiveArtifacts 'model1'
+				archiveArtifacts 'model1.h5'
 			}
 		}
     }
diff --git a/train.py b/train.py
index e6c2479..9c43cd7 100755
--- a/train.py
+++ b/train.py
@@ -39,4 +39,4 @@ model = regression_model()
 model.fit(x_train, y_train, epochs = 600, verbose = 1)
 
 y_pred = model.predict(x_test)
-model.save('model1')
\ No newline at end of file
+model.save('model1.h5')
\ No newline at end of file