s434784/ium_434784
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Maciej Sobkowiak 2021-05-16 19:41:13 +02:00
parent 1c820d6b2a
commit 6bba24a4b9
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -1,3 +1,3 @@
__pycache__
.ipynb_checkpoints
sucides_model.h5
suicide_model.h5

2
train.Jenkinsfile
View File

@ -40,7 +40,7 @@ pipeline {
}
stage('archive-model') {
steps{
archiveArtifacts 'sucides_model.h5'
archiveArtifacts 'suicide_model.h5'
}
}
stage('send-mail') {

2
training.py
View File

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