From 6bba24a4b94792da5ac6a209f8ab6334fb17c3c7 Mon Sep 17 00:00:00 2001 From: Maciej Sobkowiak Date: Sun, 16 May 2021 19:41:13 +0200 Subject: [PATCH] fix --- .gitignore | 2 +- train.Jenkinsfile | 2 +- training.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index 5685412..d1b05b3 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,3 @@ __pycache__ .ipynb_checkpoints -sucides_model.h5 \ No newline at end of file +suicide_model.h5 \ No newline at end of file diff --git a/train.Jenkinsfile b/train.Jenkinsfile index ad1441d..9ff0ddd 100644 --- a/train.Jenkinsfile +++ b/train.Jenkinsfile @@ -40,7 +40,7 @@ pipeline { } stage('archive-model') { steps{ - archiveArtifacts 'sucides_model.h5' + archiveArtifacts 'suicide_model.h5' } } stage('send-mail') { diff --git a/training.py b/training.py index b06499e..8ecbbb1 100644 --- a/training.py +++ b/training.py @@ -91,4 +91,4 @@ test_predictions = model.predict(X_test).flatten() predictions = model.predict(X_test) pd.DataFrame(predictions).to_csv('results.csv') -model.save('sucides_model.h5') +model.save('suicide_model.h5')