bluecast.preprocessing.onehot_encoding

Module Contents

Classes

OneHotCategoryEncoder

Onehot encode categorical features.

class bluecast.preprocessing.onehot_encoding.OneHotCategoryEncoder(cat_columns: List[str | float | int], target_col: str | float | int)

Onehot encode categorical features.

fit_transform(x: pandas.DataFrame, y: pandas.Series) → pandas.DataFrame

Fit onehot encoder and transform column.

transform(x: pandas.DataFrame) → pandas.DataFrame

Transform categories based on already trained encoder.

append_encoded_columns(x: pandas.DataFrame, encoded_cats: pandas.DataFrame) → pandas.DataFrame

Append encoded columns to the DataFrame.