Chapter 09
load model
NotebookPython 3 (ipykernel)9 cells
In [3]python · cell 1
python
import mlflow
TRACKING_URL = "http://127.0.0.1:5000"
mlflow.set_tracking_uri(TRACKING_URL)In [5]python · cell 2
python
run_id = '6dd459b11b4e48dc862f4e1019d166f6'
logged_model = f'runs:/{run_id}/model'
loaded_model = mlflow.pyfunc.load_model(logged_model)In [1]python · cell 3
python
import mlflowIn [2]python · cell 4
python
loaded_model = mlflow.pyfunc.load_model('s3://mlflow-models-alexey/1/6dd459b11b4e48dc862f4e1019d166f6/artifacts/model/')In [ ]python · cell 5
python
In [3]python · cell 6
python
request = {
'lpep_pickup_datetime': '2021-01-01 00:15:56',
'PULocationID': 43,
'DOLocationID': 151,
'passenger_count': 1.0,
'trip_distance': 1.01
}In [4]python · cell 7
python
features = {}
features['PU_DO'] = '%s_%s' % (request['PULocationID'], request['DOLocationID'])
features['trip_distance'] = request['trip_distance']In [5]python · cell 8
python
loaded_model.predict(features)Output
array([14.8285286])
In [ ]python · cell 9
python