diff --git a/lab8/Jenkinsfile.mlflow b/lab8/Jenkinsfile.mlflow new file mode 100644 index 0000000..56e7bb7 --- /dev/null +++ b/lab8/Jenkinsfile.mlflow @@ -0,0 +1,16 @@ +pipeline { + agent { + docker { + image 'mikolajk/ium:latest' + args '-v /mlruns:/mlruns' + } + } + stages { + stage('Stage') { + steps { + sh 'cd lab8' + sh "python predictMlflow.py" + } + } + } +} \ No newline at end of file diff --git a/lab8/predictMlflow.py b/lab8/predictMlflow.py new file mode 100644 index 0000000..fe17266 --- /dev/null +++ b/lab8/predictMlflow.py @@ -0,0 +1,15 @@ +import mlflow +import numpy as np +import json + +logged_model = '/mlruns/14/80fe21a0804844088147d15a3cebb3e5/artifacts/lego-model' + +# Load model as a PyFuncModel. +loaded_model = mlflow.pyfunc.load_model(logged_model) + +with open(f'{loaded_model}/input_example.json') as f: + input_example_data = json.load(f) + + # Predictions + print(f'input: {input_example_data}') + print(f'predictions: {loaded_model.predict(input_example_data)}') \ No newline at end of file