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.