tabensemb.trainer.Trainer.plot_scatter#
method
- Trainer.plot_scatter(x_col: str, y_col: str, category: str | None = None, ax=None, clr: Iterable | None = None, imputed: bool = False, kde_color: bool = False, figure_kwargs: Dict | None = None, scatter_kwargs: Dict | None = None, select_by_value_kwargs: Dict | None = None, savefig_kwargs: Dict | None = None, legend_kwargs: Dict | None = None, save_show_close: bool = True) Axes[source]#
Plot one column against another.
- Parameters:
- x_col
The column for the x-axis.
- y_col
The column for the y-axis.
- category
The category to classify data points with different colors and markers.
- ax
matplotlib.axes.Axes- clr
A seaborn color palette or an Iterable of colors. For example seaborn.color_palette(“deep”).
- imputed
Whether the imputed dataset should be considered.
- kde_color
Whether the scatters are colored by their KDE density.
- figure_kwargs
Arguments for
plt.figure.- scatter_kwargs
Arguments for
plt.scatter()- select_by_value_kwargs
Arguments for
tabensemb.data.datamodule.DataModule.select_by_value().- savefig_kwargs
Arguments for
plt.savefig- legend_kwargs
Arguments for
plt.legend- save_show_close
Whether to save, show (in the notebook), and close the figure if
axis not given.
- Returns:
- matplotlib.axes.Axes