Use this tag for questions about the machine-learning platform MLflow
Questions tagged [mlflow]
722 questions
3
votes
1 answer
How do I save a yolov5 model among the registered models in MLflow and then load it either directly or from weights (.pt) to make predictions?
My objective is to :
save the yolov5 model in MLflow Model Registry
Load the model directly from model registry so that we can make predictions.
Have access to the results of prediction (bounding boxes).
Current steps followed:
Cloned the yolov5…
atharva parikh
- 41
- 3
3
votes
1 answer
How to load model from mlflow with custom predict without local file?
I want to log model with custom predict. Example of signature
from sklearn.ensemble import RandomForestRegressor
class CustomForest(RandomForestRegressor):
def predict(self, X, second_arg=False):
pred = super().predict(X)
value =…
Nourless
- 729
- 1
- 5
- 18
3
votes
1 answer
'PythonModelContext' object returned from mlflow.pyfunc.load_model - how to retrieve original model that was saved
I am creating a custom myflow.pyfunc object that I would like to save to MLFlow and retrieve later. I don't understand the relationship between the object that is saved with mlflow.pyfunc.save_model(), and the one that is retrieved with…
zmek
- 53
- 7
3
votes
1 answer
AttributeError: module 'collections' has no attribute 'MutableSet'
I've written an MLflow component to upload & log an artifact from a URL to WandB. I am trying to run the script inside a conda environment but unfortunately I am facing this error
File…
Prashant singh
- 41
- 1
- 3
3
votes
1 answer
Read csv files in a MLFlow pipeline
I'm following this documentation to use ML Flow pipelines, which requires to clone this repository. If I run the complete pipeline as it is It works perfectly:
import os
from mlflow.pipelines import…
Luis Ramon Ramirez Rodriguez
- 9,591
- 27
- 102
- 181
3
votes
0 answers
Pins + Vetiver vs MLflow which one to choose for MLOps
I am a big fan boy of tidymodels and played around with vetiver + pins in R and Python in order to not only develop models but actually deploy them.
However, if you are looking for tools that support in the area of MLOps, sooner or later you will…
Mischa
- 137
- 8
3
votes
3 answers
MLflow proxied artifact access: Unable to locate credentials
I am using MLflow to track my experiments. I am using an S3 bucket as an artifact store. For acessing it, I want to use proxied artifact access, as described in the docs, however this does not work for me, since it locally looks for credentials (but…
bk_
- 751
- 1
- 8
- 27
3
votes
1 answer
How can I save loss and accuracy metrics in mlflow after each epoch?
I would like to see metrics like loss and accuracy as a graph by storing each value for the corresponding metrics after each epoch during training/testing phase of a keras model.
PS: I know that we can do it by using autolog feature of mlflow for…
Manas
- 888
- 10
- 20
3
votes
1 answer
How to log model using mlflow REST api? Does mlflow REST APIs support it?
I'm writing a library using mlflow REST APIs.
I'm looking for mlflow REST api for logging different mlflow models.
In the doc, https://www.mlflow.org/docs/latest/rest-api.html#log-model it says the api will be removed in future and doesn't have…
veeresh patil
- 1,168
- 1
- 11
- 18
3
votes
1 answer
Multiple values for a single parameter in the mlflow run command
I just started learning mlflow and wanted to know how to pass multiple values to each parameter in the mlflow run command.
The objective is to pass a dictionary to GridSearchCV as a param_grid to perform cross validation.
In my main code, I…
Downforu
- 317
- 5
- 13
3
votes
0 answers
How to configure artifact store of mlflow tracking service to connect to minio S3 using minio STS generated acces_key, secret_key and session_token
Minio is configured with LDAP and am generating credentials of user
with AssumeRoleWithLDAPIdentity using STS API (reference)
From above values, I'm setting the variables AWS_ACCESS_KEY, AWS_SECRET_KEY, AWS_SESSION_TOKEN (reference)
I'm getting…
veeresh patil
- 1,168
- 1
- 11
- 18
3
votes
1 answer
Difference between tracking_uri and the backend store uri in MLFLOW
I am using Mlflow for my project hosting it in an EC2 instance. I was wondering in MlFlow what is the difference between the backend_store_uri we set when we launch the server and the trarcking_uri ?
Thanks,
Ali Tber
- 33
- 4
3
votes
2 answers
MLflow saves models to relative place instead of tracking_uri
sorry if my question is too basic, but cannot solve it.
I am experimenting with mlflow currently and facing the following issue:
Even if I have set the tracking_uri, the mlflow artifacts are saved to the ./mlruns/... folder relative to the path from…
László Marton
- 65
- 7
3
votes
0 answers
Log experiments with MLflow to DAGsHub tracking server
I'm trying to use MLflow and log my experiments to DAGsHub's remote-tracking server but I get this error message:
WARNING mlflow.utils.autologging_utils: Encountered unexpected error during tensorflow autologging: 'NoneType' object has no attribute…
ParrisNevada
- 53
- 3
3
votes
2 answers
Log Pickle files as a part of Mlflow run
I am running an MLflow experiment as a part of it I would like to log a few artifacts as a python pickle.
Ex: Trying out different categorical encoders, so wanted to log the encoder objects as a pickle file.
Is there a way to achieve this?
Naga Budigam
- 689
- 1
- 10
- 26