Use this tag for questions about the machine-learning platform MLflow
Questions tagged [mlflow]
722 questions
3
votes
1 answer
How can I use Hyperopt with MLFlow within a pandas_udf?
I'm building multiple Prophet models where each model is passed to a pandas_udf function which trains the model and stores the results with MLflow.
@pandas_udf(result_schema, PandasUDFType.GROUPED_MAP)
def forecast(data):
......
with…
nescobar
- 81
- 1
- 5
3
votes
1 answer
How to serve custom MLflow model with Docker?
We have a project following essentially this
docker example with the only difference that we created a custom model similar to this whose code lies in a directory called forecast. We succeeded in running the model with mlflow run. The problem arises…
bruco
- 141
- 1
- 12
3
votes
1 answer
Error while running a local mlflow server
I am trying to run an mlflow server locally.
To do so, I am using:
mlflow server --backend-store-uri="sqlite:///C:\\path\\to\\project_folder\\backend\\mlflow_data.db"
…
quant
- 4,062
- 5
- 29
- 70
3
votes
0 answers
adbazureml not supported by mlflow
We've been following the latest Microsoft webinar about deploying the ML model from Azure Databricks to Azure ML using the MLFlow, and we get the following error when trying to run the experiment from Databricks notebook using the following…
Boris Kr
- 31
- 1
3
votes
1 answer
How to integrate mlflow with tensorflow object detection api
I am trying to use the mlflow.tensorflow.autolog() with tensorflow object detection api. Adding mlflow.tensorflow.autolog() to model_main.py logs some parameters like…
zishan ahmed
- 101
- 1
- 9
3
votes
1 answer
Saving artifacts on remote mlflow server
I am trying to store MLflow artifacts on a remote server running MLflow. The server I am accessing from and server running MLflow are both VMs on google cloud. I can see the matrices in the MLflow server but not the artifacts.
I tried the flollowing…
bazinga
- 2,120
- 4
- 21
- 35
3
votes
2 answers
How do I set a custom gunicorn worker timeout when serving an MLflow model with the "mlflow models serve" CLI?
When serving an MLflow Python model with the "pyfunc" backend (https://github.com/mlflow/mlflow/blob/master/mlflow/pyfunc/backend.py), how can I set a custom gunicorn worker timeout? The default timeout of 60 seconds may be insufficient when serving…
Corey Zumar
- 81
- 1
- 4
3
votes
0 answers
Assume IAM Role to store MLFlow Artifact on S3 bucket in another account
I have to save my MLFlow artifacts (using Databricks Unified Analytics) to a S3 bucket, with service-side encrpytion using a KMS key.
My instances are into an AWS account A, my S3 bucket and my KMS key into an account B. I can't have my KMS Key into…
Behel
- 31
- 1
3
votes
2 answers
Unable to run mlflow ui in Jupyter
I am new to MLflow. I was trying to use it in Jupyter. As part of the quickstart, I ran the following code:
import os
from mlflow import log_metric, log_param, log_artifact
if __name__ == "__main__":
# Log a parameter (key-value pair)
…
Derek Langley
- 172
- 1
- 11
3
votes
0 answers
Deploying ml model using azureml and mlflow
I'm new to databricks and deploying models using mlflow and azureml, I'm trying to deploy my model but haven't found a lot of documentation or examples.
I have my model which I save using:
mlflow.sklearn.save_model(model, model_path,
…
Daniel Zapata
- 812
- 3
- 9
- 31
3
votes
1 answer
MLFlow Projects throw JSONDecode error when run
I'm trying to get MLFlow Projects to run using the MLFlow CLI and its following the tutorial leads to an error. For any project I try to run from the CLI, I get the following error
Traceback (most recent call last):
File…
AustenStewart
- 31
- 3
2
votes
0 answers
is ML Flow Pytorch Lightning autologging supposed to automatically log metrics?
I am attempting to use MLflow to log a pytorch lightning model to experiments in Databricks. Accoring to the documentation, it seems like metrics like training and validation loss are supposed to be logged automatically without having to call…
Mikel
- 21
- 1
2
votes
1 answer
Setting tags during model logging mlflow
I am logging the model using
mlflow.sklearn.log_model(model, "my-model")
and I want to set tags to the model during logging, I checked that this method does not allow to set tags, there is a mlflow.set_tags() method but it is tagging the run not…
Eddmik
- 155
- 7
2
votes
0 answers
Load ML-Model from mlruns folder via GitHub
On my local machine, I have tuned and logged the hyperparams for a ML-model and created a public GitHub Repo with the mlruns folder.
Because I also want to use the model in a Google Collab environment, I am trying to figure out how to load that…
AnonymUser
- 45
- 4
2
votes
0 answers
MLFlow - How to configure TrackingServer's logs (NOT Log experiment data)
I'm attempting to diagnose issues with an AWS deployed container instance of MLFlow's tracking server.
To ensure I've not misconfigured a database, s3 bucket or other system I'd like to see output from a logger such as this…
NWTRP
- 21
- 1