This commit is contained in:
Dominik Strzako 2021-05-27 13:15:32 +02:00
parent d7f5dc3aec
commit f10aeb31a0
2 changed files with 2 additions and 1 deletions

View File

@ -20,6 +20,7 @@ pipeline {
{
sh 'rm -r my_model'
copyArtifacts(fingerprintArtifacts: true, projectName: 's434788-training/mlflow', selector: buildParameter('WHICH_BUILD'))
sh 'ls'
}
}
stage('predict')

View File

@ -19,7 +19,7 @@ model = mlflow.keras.load_model(
model_uri=f"models:/{model_name}/{version}"
)
data = pd.read_json('my_model/input_example.json', orient='split')
data = pd.read_json('my_model/input_example.json', orient='index')
print(data)
print(model.predict(data))