tabensemb.trainer.Trainer._plot_action_category_unique_values#

method

Trainer._plot_action_category_unique_values(df: DataFrame, category: str) Tuple[Series, ndarray][source]#

Get the category to classify data points and its unique values.

Parameters:
df

The dataframe. The returned Series has the same indices.

category

The category to classify data points.

Returns:
pd.Series

The category

np.ndarray

Unique values.