tabensemb.trainer.Trainer._plot_action_categorical_scatter#

method

Trainer._plot_action_categorical_scatter(x, y, df: DataFrame, category: str, ax, clr: Iterable, scatter_kwargs: Dict)[source]#

Plot scatters whose colors are related to their category.

Parameters:
x

x-values of the scatter plot.

y

y-values of the scatter plot.

df

The dataframe whose category column is used to classify data points.

category

The column to classify data points.

ax

matplotlib.axes.Axes

clr

A seaborn color palette or an Iterable of colors. For example seaborn.color_palette(“deep”).

scatter_kwargs

Arguments for plt.scatter