Questions tagged [mlflow]

7 questions
8
votes
1 answer

MLflow real world experience

Can someone provide a summary of the real world deployment experience of MLflow? We have a few ML models (e.g., LightGBM, tensorflow v2, etc.) and want to avoid framework like SageMaker (due to customer requirement). So we are looking into various…
David293836
  • 197
  • 1
  • 6
1
vote
1 answer

Multiple values for a single parameter in the mlflow run command

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 retrieve the command line parameters using argparse. And…
Downforu
  • 131
  • 3
0
votes
1 answer

Can't load custom Keras metrics using mlflow.pyfunc

I have a DNN in Keras, which includes a custom metric function and which I want to pipeline with some SKlearn preprocessing. I further want to persist the model using MLFlow for easy deployment. The requirement to pipeline with sklearn means that I…
Dan Scally
  • 1,724
  • 6
  • 23
0
votes
0 answers

Is the MLFlow platform the only platform for reusable ML? Is it necessarily the best?

Is the MLFlow platform the only platform for reusable ML? Is it necessarily the best (yes it can be subjective and case-dependent)? I found at least one paper that discusses this platform: Accelerating the Machine Learning Lifecycle with MLflow…
mavavilj
  • 416
  • 1
  • 3
  • 12
0
votes
0 answers

How to prevent MLFlow from automatically creating a run when a logging function is called?

I have a custom model class that calls mlflow.llm.log_predictions in the predict method like so: class Model: ... def predict(input) -> Output: ... mlflow.llm.log_predictions(...) ... I'm using that predict method…
0
votes
0 answers

How to convert ndarray to spark dataframe for mlflow prediction?

I have a trained model, and test input from mnist import mlflow from pyspark.sql.functions import struct, col logged_model = 'runs:/myid/myModel' loaded_model = mlflow.pyfunc.spark_udf(spark, model_uri=logged_model,…
olaf
  • 101
  • 1
0
votes
1 answer

Store preprocessing function along with model in mlflow.keras

The following is a simplified code snipet that is relevant to storing keras LSTM models in MLFlow. with mlflow.start_run() as run: mlflow.keras.log_model(model,"lstm") mlflow.log_params(model_parameters) …
Enk9456
  • 73
  • 1
  • 11