.
This commit is contained in:
parent
7d0bc87876
commit
c379818a13
@ -26,12 +26,8 @@ pipeline {
|
||||
stage('evaluation') {
|
||||
steps {
|
||||
script {
|
||||
def image = docker.build('dock')
|
||||
image.inside{
|
||||
sh 'chmod +x evaluation.py'
|
||||
sh 'python3 evaluation.py'
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
|
||||
from keras import backend as K
|
||||
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')
|
||||
# test = pd.read_csv('test.csv')
|
||||
|
@ -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('suicide_model.h5')
|
||||
model.save('suicide_model')
|
||||
|
Loading…
Reference in New Issue
Block a user