tabensemb.trainer.Trainer.plot_partial_dependence_2way_all

method

Trainer.plot_partial_dependence_2way_all(program: str, model_name: str, x_feature: str, y_features: List[str] | None = None, 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 2-way partial dependence plots with bootstrapping. One continuous feature is fixed for x-axis. The rest of the continuous features are on y-axis, respectively.

Parameters:

program

The selected model base.

model_name

The selected model in the model base.

x_feature

The continuous feature fixed for x-axis.

y_features

Continuous features on y-axis respectively. If None, all other continuous features are used.

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_2way().

Returns:

matplotlib.figure.Figure

The figure if save_show_close is False.