atm.encoder module

Classes

DataEncoder([class_column, feature_columns])

class atm.encoder.DataEncoder(class_column='class', feature_columns=None)

Bases: object

Methods

fit(data)	Fit one-hot encoders for categorical features and an integer encoder for the label.
fit_transform(data)	Process data into a form that ATM can use.
transform(data)	Convert a DataFrame of labeled data to a feature matrix in the form that ATM can use.

fit(data)

Fit one-hot encoders for categorical features and an integer encoder for the label. These can be used later to transform raw data into a form that ATM can work with.

data: pd.DataFrame of unprocessed data

fit_transform(data)

Process data into a form that ATM can use.

transform(data)

Convert a DataFrame of labeled data to a feature matrix in the form that ATM can use.