tabensemb.trainer.Trainer.plot_partial_dependence_all

method

Trainer.plot_partial_dependence_all(program: str, model_name: str, fontsize=12, figure_kwargs: Dict | None = None, get_figsize_kwargs: Dict | None = None, savefig_kwargs: Dict | None = None, save_show_close: bool = True, tqdm_active: bool = False, **kwargs) → None | Figure

Calculate and plot partial dependence plots with bootstrapping.

Parameters:

program

The selected model base.

model_name

The selected model in the model base.

fontsize

plt.rcParams["font.size"]

figure_kwargs

Arguments for plt.figure.

get_figsize_kwargs

Arguments for tabensemb.utils.utils.get_figsize().

savefig_kwargs

Arguments for plt.savefig

save_show_close

Whether to save, show (in the notebook), and close the figure, or return the matplotlib.figure.Figure instance.

tqdm_active

Whether to use a tqdm progress bar.

kwargs

Arguments for plot_partial_dependence().

Returns:

matplotlib.figure.Figure

The figure if save_show_close is False.