Use this tag for questions about the machine-learning platform MLflow
Questions tagged [mlflow]
722 questions
2
votes
0 answers
Reuse existing conda env for mlflow project?
I have configured a mlflow project file. First hard knock was that the extension is not required. The current problem is that I have exported an existing conda environment using:
conda env export --name ENVNAME > envname.yml
substituting the…

skrishnan_v
- 31
- 4
2
votes
1 answer
How to run non-spark model training task (using fasttext) efficiently on a databricks cluster?
I want to train some models using fasttext and since it doesn't use spark, it will be running on my driver. The number of training jobs that will be running simultaneously is very large and so is the size of the data. Is there a way to make it run…

BBloggsbott
- 195
- 2
- 12
2
votes
1 answer
Can I change the port of my MLflow tracking server?
I would like to know if I can change the port of my MLflow server.
By default it is running on port 5000, but my company's VPN only allows HTTP (port 80) and HTTPS (port 443) traffic.
This might be a very beginner's question, but is it possible, and…

yoyoog
- 255
- 1
- 10
2
votes
2 answers
How to set a tag at the experiment level in MLFlow
I can see that an experiment in MLFlow can have tags (like runs can have tags).
I'm able to set a run's tag using mlflow.set_tag, but how do I set it for an experiment?

Nestor
- 13,706
- 11
- 78
- 119
2
votes
0 answers
Cannot log sklearn model to minio (s3) server
My docker-composed setup consists of 3 containers - mlflow + minio(s3) + mysql and I am trying to run a simple sklearn training model python script from within the mlflow container and it works fine but for the last line below -
import pandas as…

ZeroGraviti
- 1,047
- 2
- 12
- 28
2
votes
0 answers
Error can't get attribute Net when saving PyTorch model with MLFlow
After installing MLFlow using one-click-mlflow I save a pytorch model using the default command that I found in the user guide. You can find the command bellow:
mlflow.pytorch.log_model(net, artifact_path="model", pickle_module=pickle)
The neural…

ucsky
- 442
- 6
- 13
2
votes
2 answers
Pyspark: How to save and apply IndexToString to convert labels back to original values in a new predicted dataset
I am using pyspark.ml.RandomForestClassifier and one of the steps here involves StringIndexer on the training data target variable to convert it into labels.
indexer = StringIndexer(inputCol = target_variable_name, outputCol = 'label').fit(df)
df =…

Deb
- 499
- 2
- 15
2
votes
1 answer
Hi. I am very new to MLFlow, and want to implement MLFlow project on my own ML model. However I am getting ""Could not find main among entry points""
The full error message is as below:
ERROR mlflow.cli: === Could not find main among entry points [] or interpret main as a runnable script. Supported script file extensions: ['.py', '.sh'] ===
I also try the solutions suggested here…

nihad1918
- 33
- 6
2
votes
1 answer
MLFlow doesnt copy properly my artifacts to the mlruns folder when used in a container
I am logging a few models using MLflow. The corresponding scripts are executed in a docker container.
The command mlflow.log_artifacts doesn't seem to work properly since I can t see the corresponding files under the mlruns folder. The command…

greg hor
- 682
- 6
- 17
2
votes
0 answers
MLFLOW Artifacts stored on ftp server but not showing in ui
I use MLFLOW to store some parameters and metrics during training on a remote tracking server. Now I tried to also add a .png file as an artifact, but since the MLFLOW server is running remotely I store the file on a ftp server. I gave the ftp…

Wildner
- 21
- 2
2
votes
2 answers
How can I retrive the model.pkl in the experiment in Databricks
I want to retrieve the pickle off my trained model, which I know is in the run file inside my experiments in Databricks.
It seems that the mlflow.pyfunc.load_model can only do the predict method.
There is an option to directly access the pickle?
I…

Leonardo Kanashiro Felizardo
- 240
- 2
- 17
2
votes
1 answer
MLflow Rstudio Dockerfile Permission denied
Goal
Track R models in the remote MLflow tracking server (running in kubernetes). The model is developed on the local computer from RStudio run in the Docker container.
Setup
Based on my research I need to create RStudio image with conda installed.…

Michal Hucko
- 357
- 5
- 18
2
votes
2 answers
Copying Experiments from a MLFlow server to another MLFlow server
I have a user in a Linux machine and I run a mlflow server from this user. Artifacts are stored in local mlruns folder. Lets call this user as user A. Then I run another mlflow server from another Linux user and call this user as user B. I wanted to…

tkarahan
- 315
- 1
- 2
- 15
2
votes
0 answers
Log (and then apply) Spark MLlib model from R to MLflow
I'm using Spark MLlib functions (through the sparklyr package) to train a model but now seem unable to save the model in MLflow for future use.
iris_tbl <- sparklyr::copy_to(sc, iris, "iris_spark")
mdl_mllib <- iris_tbl %>%…

josephD
- 142
- 7
2
votes
0 answers
Is there any way to log MLflow model coming from tensorflow.keras using custom model wrapper?
I am trying to log a Tensorflow.keras model using mlflow. In my workflow I need to define a custom .predict() . So based on this I have done these steps:
First I trained the tf.keras model as following:
_input = (
…

Àlex Encalado Masià
- 56
- 4