sklearn_dummies API¶

Base module.

class sklearn_dummies.base.DataFrameDummies¶

Bases: sklearn.base.TransformerMixin

cat_cols¶

List of categorical columns

Type:	list

final_cols¶

List of all columns with dummy values

Type:	list

fit(df, y=None)¶

Parameters:	df (pd.DataFrame) – Provides the column names to be used.
Returns:	Itself.
Return type:	DataFrameDummies

transform(df, y=None)¶

Parameters:	df (pd.DataFrame) – Data to be dummified.
Returns:	Transformed data
Return type:	pd.DataFrame

class sklearn_dummies.base.NPArrayDummies¶

Bases: sklearn.base.TransformerMixin

labels¶

List of labels

Type:	list

fit(X, y=None)¶

Parameters:	X (np.ndarray) – Provides the labels.
Returns:	Itself.
Return type:	NPArrayDummies

transform(X, y=None)¶

Parameters:	X (np.ndarray) – Data to be dummified.
Returns:	Xt – Transformed data
Return type:	np.ndarray