This commit is contained in:
Dominik Strzako 2021-05-23 22:19:08 +02:00
parent ca2841a2e4
commit 5f656b2816
2 changed files with 5 additions and 6 deletions

View File

@ -13,7 +13,7 @@ pipeline {
{
steps
{
copyArtifacts(fingerprintArtifacts: true, projectName: 's434704-training/master', selector: buildParameter('WHICH_BUILD'))
copyArtifacts(fingerprintArtifacts: true, projectName: 's434695-training/train', selector: buildParameter('WHICH_BUILD'))
}
}
stage('predict')

View File

@ -1,9 +1,8 @@
import mlflow
import sys
import json
import pandas as pd
model = mlflow.keras.load_model("my_model")
data = pd.read_json('my_model/input_example.json', orient='index')
model = mlflow.keras.load_model("movies_on_streaming_platforms_model")
with open('movies_on_streaming_platforms_model/input_example.json') as json_file:
data = json.load(json_file)
print(data)
print(model.predict(data))