OneHotGroupGenerator

class mastml.feature_generators.OneHotGroupGenerator(groups, remove_constant_columns=False)

Bases: BaseGenerator

Class to generate one-hot encoded values from a list of categories using scikit-learn’s one hot encoder method More info at: https://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.OneHotEncoder.html

Args:

groups: (pd.Series): pandas Series of group (category) names

remove_constant_columns: (bool), whether to remove constant columns from the generated feature set

Methods:

fit: pass through, copies input columns as pre-generated features

Args:

X: (pd.DataFrame), input dataframe containing X data

y: (pd.Series), series containing y data

transform: generate the one-hot encoded features. There will be n columns made, where n = number of unique categories in groups

Args:

None.

Returns:

df: (dataframe), output dataframe containing generated features

y: (series), output y data as series

Methods Summary

fit(X[, y])
transform([X])

Methods Documentation

fit(X, y=None)

transform(X=None)