Skip to content

plot_timeline

method plot_timeline(models=None, title=None, legend="lower right", figsize=(900, 600), filename=None, display=True)[source]
Plot the timeline of a study.

This plot is only available for models that ran hyperparameter tuning.

Parametersmodels: int, str, Model, slice, sequence or None, default=None
Models to plot. If None, all models that used hyperparameter tuning are selected.

title: str, dict or None, default=None
Title for the plot.

legend: str, dict or None, default="lower right",
Legend for the plot. See the user guide for an extended description of the choices.

  • If None: No legend is shown.
  • If str: Location where to show the legend.
  • If dict: Legend configuration.

figsize: tuple, default=(900, 600)
Figure's size in pixels, format as (x, y)

filename: str or None, default=None
Save the plot using this name. Use "auto" for automatic naming. The type of the file depends on the provided name (.html, .png, .pdf, etc...). If filename has no file type, the plot is saved as html. If None, the plot is not saved.

display: bool or None, default=True
Whether to render the plot. If None, it returns the figure.

Returnsgo.Figure or None
Plot object. Only returned if display=None.


See Also

plot_edf

Plot the Empirical Distribution Function of a study.

plot_slice

Plot the parameter relationship in a study.

plot_terminator_improvement

Plot the potentials for future objective improvement.


Example

>>> from atom import ATOMClassifier
>>> from optuna.pruners import PatientPruner

>>> X = pd.read_csv("./examples/datasets/weatherAUS.csv")

>>> atom = ATOMClassifier(X, y="RainTomorrow", n_rows=1e4)
>>> atom.impute()
>>> atom.encode()
>>> atom.run(
...     models="LGB",
...     n_trials=15,
...     ht_params={"pruner": PatientPruner(None, patience=100)},
... )
>>> atom.plot_timeline()