diff --git a/Jenkinsfile-from-registry b/Jenkinsfile-from-registry new file mode 100644 index 0000000..147b0ce --- /dev/null +++ b/Jenkinsfile-from-registry @@ -0,0 +1,18 @@ +pipeline { + agent { + dockerfile true + } + + stage('predict') + { + steps + { + catchError { + sh 'ls -al' + sh 'python3 predict434788-from-registry.py' + } + } + } + + } +} diff --git a/predict434788-from-registry.py b/predict434788-from-registry.py new file mode 100644 index 0000000..566833d --- /dev/null +++ b/predict434788-from-registry.py @@ -0,0 +1,20 @@ +from mlflow.tracking import MlflowClient +import mlflow +import pandas as pd + +mlflow.set_tracking_uri("http://172.17.0.1:5000") +client = MlflowClient() +version = 0 +model_name = "s434788" +for mv in client.search_model_versions(f"name='{model_name}'"): + if int(mv.version) > version: + version = int(mv.version) + +model = mlflow.keras.load_model( + model_uri=f"models:/{model_name}/{version}" +) + +data = pd.read_json('my_model/input_example.json', orient='index') + +print(data) +print(model.predict(data)) \ No newline at end of file