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')