tabensemb.trainer.Trainer.plot_corr#
method
- Trainer.plot_corr(fontsize: Any = 10, imputed=False, features: List[str] | None = None, method: str | Callable = 'pearson', include_label: bool = True, ax=None, figure_kwargs: Dict | None = None, imshow_kwargs: Dict | None = None, select_by_value_kwargs: Dict | None = None, savefig_kwargs: Dict | None = None, save_show_close: bool = True) Axes[source]#
Plot correlation coefficients among features and the target.
- Parameters:
- fontsize
The
fontsizeargument for matplotlib.- imputed
Whether the imputed dataset should be considered. If False, some NaN coefficients may exist for features with missing values.
- features
A subset of continuous features to calculate correlations on.
- method
The argument of
pd.DataFrame.corr. “pearson”, “kendall”, “spearman” or Callable.- include_label
If True, the target is also considered.
- ax
matplotlib.axes.Axes- figure_kwargs
Arguments for
plt.figure.- imshow_kwargs
Arguments for
plt.imshow.- select_by_value_kwargs
Arguments for
tabensemb.data.datamodule.DataModule.select_by_value().- savefig_kwargs
Arguments for
plt.savefig- save_show_close
Whether to save, show (in the notebook), and close the figure if
axis not given.
- Returns:
- matplotlib.axes.Axes