From 1d06ce26b834667a2ab47c832e8dd69c97a12520 Mon Sep 17 00:00:00 2001 From: Filip Izydorczyk Date: Tue, 8 Jun 2021 18:32:35 +0200 Subject: [PATCH] predict from registry test --- predict-registry/Jenkinsfile | 22 ++++++++++++++++++++++ predict-registry/predict.py | 14 ++++++++++++++ 2 files changed, 36 insertions(+) create mode 100644 predict-registry/Jenkinsfile create mode 100644 predict-registry/predict.py diff --git a/predict-registry/Jenkinsfile b/predict-registry/Jenkinsfile new file mode 100644 index 0000000..a0dff30 --- /dev/null +++ b/predict-registry/Jenkinsfile @@ -0,0 +1,22 @@ +pipeline { + agent {dockerfile true} + + + stages { + stage('Copy Archive') { + steps { + script { + step ([$class: 'CopyArtifact', + projectName: 's430705-training/master', + target: 'prediction_data']) + } + } + } + stage('Predict example value'){ + steps { + sh "cd /tmp" + sh "ls -al" + } + } + } +} \ No newline at end of file diff --git a/predict-registry/predict.py b/predict-registry/predict.py new file mode 100644 index 0000000..dc3daa3 --- /dev/null +++ b/predict-registry/predict.py @@ -0,0 +1,14 @@ +import json +import mlflow +import pandas as pd +from pprint import pprint +from mlflow.tracking import MlflowClient + +model_name = "s430705" +model_version = 50 + +mlflow.set_tracking_uri("http://172.17.0.1:5000") + +model = mlflow.keras.pyfunc.load_model( + model_uri=f"models:/{model_name}/{model_version}", +)