Вы не можете выбрать более 25 тем
Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.
18 строки
553 B
Python
18 строки
553 B
Python
import mlflow
|
|
import pickle as pkl
|
|
|
|
# Работаем с MLflow локально
|
|
TRACKING_SERVER_HOST = "127.0.0.1"
|
|
TRACKING_SERVER_PORT = 5001
|
|
|
|
registry_uri = f"http://{TRACKING_SERVER_HOST}:{TRACKING_SERVER_PORT}"
|
|
tracking_uri = f"http://{TRACKING_SERVER_HOST}:{TRACKING_SERVER_PORT}"
|
|
|
|
mlflow.set_tracking_uri(tracking_uri)
|
|
mlflow.set_registry_uri(registry_uri)
|
|
|
|
RUN_NAME = '96a46920d98c48dfa3c019926b44018b'
|
|
loaded_model = mlflow.sklearn.load_model(f'runs:/{RUN_NAME}/models')
|
|
|
|
with open('model.pkl', 'wb+') as f:
|
|
pkl.dump(loaded_model, f) |