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

.
Some checks failed
s434784-evaluation/pipeline/head There was a failure building this commit
Details
s434784-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Maciej Sobkowiak 2021-05-16 22:14:09 +02:00
parent 7d0bc87876
commit c379818a13
3 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
eval.Jenkinsfile
View File

@ -26,12 +26,8 @@ pipeline {
stage('evaluation') { stage('evaluation') {
steps { steps {
script { script {
def image = docker.build('dock')
image.inside{
sh 'chmod +x evaluation.py' sh 'chmod +x evaluation.py'
sh 'python3 evaluation.py' sh 'python3 evaluation.py'
}
} }
} }
} }

2
evaluation.py
View File

@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
from keras import backend as K from keras import backend as K
from sklearn.metrics import mean_squared_error from sklearn.metrics import mean_squared_error
model = keras.models.load_model('suicide_model.h5') model = keras.models.load_model('suicide_model')
# train = pd.read_csv('train.csv') # train = pd.read_csv('train.csv')
# test = pd.read_csv('test.csv') # test = pd.read_csv('test.csv')

2
training.py
View File

@ -91,4 +91,4 @@ test_predictions = model.predict(X_test).flatten()
predictions = model.predict(X_test) predictions = model.predict(X_test)
pd.DataFrame(predictions).to_csv('results.csv') pd.DataFrame(predictions).to_csv('results.csv')
model.save('suicide_model.h5') model.save('suicide_model')