ModuleNotFoundError: No module named 'databricks.feature_store.mlflow_model'
Options
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
08-07-2023 05:19 PM
I'm using runtime DBR 12.2 LTS ML
Getting this error when running.
import mlflow
logged_model = 'runs:/.../model' #my run id
# Load model as a PyFuncModel.
loaded_model = mlflow.pyfunc.load_model(logged_model)
# Predict on a Pandas DataFrame.
import pandas as pd
loaded_model.predict(pd.DataFrame(data))