Skip to main content

fabric.mlflow

MLflowIntegration Objects

class MLflowIntegration()

pickle_and_log_automl_artifacts

def pickle_and_log_automl_artifacts(automl, model, estimator, signature=None)

log automl artifacts to mlflow load back with automl = mlflow.pyfunc.load_model(model_run_id_or_uri), then do prediction with automl.predict(X)

adopt_children

def adopt_children(result=None)

Set autologging child runs to nested by fetching them after all child runs are completed. Note that this may cause disorder when concurrently starting multiple AutoML processes with the same experiment name if the MLflow version is less than or equal to "2.5.0".

retrain

def retrain(train_func, config)

retrain with given config, added for logging the best config and model to parent run. No more needed after v2.0.2post2 as we no longer log best config and model to parent run.