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