Skip to content

lit_mlflow.logger

source module lit_mlflow.logger

Classes

source class DbxMLFlowLogger(experiment_name: str = 'lightning_logs', run_name: str | None = None, tracking_uri: str | None = mlflow.get_tracking_uri(), tags: dict[str, Any] | None = None, save_dir: str | None = './mlruns', log_model: Literal[True, False, 'all'] = False, prefix: str = '', artifact_location: str | None = None, run_id: str | None = None)

Bases : MLFlowLogger

Attributes

  • name : Optional[str] — Get the experiment id.

  • version : Optional[str] — Get the run id.

  • root_dir : Optional[str] — Return the root directory where all versions of an experiment get saved, or None if the logger does not save data locally.

  • log_dir : Optional[str] — Return directory the current version of the experiment gets saved, or None if the logger does not save data locally.

  • group_separator : str — Return the default separator used by the logger to group the data into subfolders.

  • save_dir : Optional[str] — The root file directory in which MLflow experiments are saved.

  • experiment_id : Optional[str] — Create the experiment if it does not exist to get the experiment id.

source property DbxMLFlowLogger.experiment: MlflowClient

source property DbxMLFlowLogger.run_id: str | None