mlflow
This commit is contained in:
parent
b19f1edd1f
commit
85fd473d61
@ -7,7 +7,7 @@ pipeline {
|
||||
}
|
||||
parameters {
|
||||
string(
|
||||
defaultValue: '{"inputs": [[0.51, 0.86]]}',
|
||||
defaultValue: '{"inputs": [[0.51, 0.86], [0.79, 0.79], [0.74, 0.77], [0.66, 0.73], [0.79, 0.47]]}',
|
||||
description: 'Input example',
|
||||
name: 'INPUT'
|
||||
)
|
||||
|
@ -6,6 +6,6 @@ model = mlflow.pyfunc.load_model('mlruns/1/4b83e774512444188fb587288818c298/arti
|
||||
|
||||
with open('input_example.json') as file:
|
||||
data = json.load(file)
|
||||
input_example = np.array([data['inputs'][0]]).reshape(-1, 2)
|
||||
input_example = np.array([data['inputs']]).reshape(-1, 2)
|
||||
print('Input: ', input_example)
|
||||
print('Predictions: ', model.predict(input_example))
|
||||
|
Loading…
Reference in New Issue
Block a user