s444409/ium_444409
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Revert file parameter

Browse Source
This commit is contained in:
Marcin Kostrzewski 2022-05-11 20:47:50 +02:00
parent 36e6ac8270
commit e9d1153a35
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Jenkinsfile-predict-s444356
View File

@ -5,10 +5,6 @@ pipeline {
}
}
parameters {
file(
name: 'input_example.json',
description: 'Example input file for the model'
)
buildSelector(
defaultSelector: lastSuccessful(),
description: 'Which build to use for copying artifacts',
@ -24,7 +20,6 @@ pipeline {
}
stage('Predict values using model from artifact') {
steps {
sh "cat input_example.json"
sh "python predict_s444356.py"
}
}

2
predict_s444356.py
View File

@ -6,7 +6,7 @@ logged_model = 'mlruns/1/296d6f314bb2451885fb7ae58988301e/artifacts/model'
loaded_model = mlflow.pyfunc.load_model(logged_model)
with open('input_example.json') as f:
with open(f'{logged_model}/input_example.json') as f:
data = json.load(f)
input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape(-1, 2)