CategoricalEncoder

class pinnx.utils.transformers.CategoricalEncoder

OneHotEncoder that can handle categorical variables.

fit(X)

Fit a list or array of categories.

Parameters:

X (array-like, shape=(n_categories,)) – List of categories.

inverse_transform(Xt)

Inverse transform one-hot encoded categories back to their original representation.

Parameters:

Xt (array-like, shape=(n_samples, n_categories)) – One-hot encoded categories.

Returns:

X – The original categories.

Return type:

array-like, shape=(n_samples,)

transform(X)

Transform an array of categories to a one-hot encoded representation.

Parameters:

X (array-like, shape=(n_samples,)) – List of categories.

Returns:

Xt – The one-hot encoded categories.

Return type:

array-like, shape=(n_samples, n_categories)