This commit is contained in:
s434695 2021-05-24 01:20:22 +02:00
parent f2e1563198
commit 1c47e59902
2 changed files with 38 additions and 0 deletions

18
Jenkinsfile-from-registry Normal file
View File

@ -0,0 +1,18 @@
pipeline {
agent {
dockerfile true
}
stage('predict')
{
steps
{
catchError {
sh 'ls -al'
sh 'python3 predict434788-from-registry.py'
}
}
}
}
}

View File

@ -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))