Skip to content

Allow more flexible run names in MLflowCallback #209

@gtauzin

Description

@gtauzin

Motivation

The naming of runs generated by MLflowCallback is not very flexible.

Description

At the moment it is the trial number (by default) or a string:

https://github.com/optuna/optuna-integration/blob/main/optuna_integration/mlflow/mlflow.py#L120C1-L120C81

It would be nice to allow the user to pass a callable that takes the trial as an input to allow full customization.

Alternatives (optional)

No response

Additional context (optional)

I am happy to take this on and submit a small PR :)

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureChange that does not break compatibility, but affects the public interfaces.

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions