prof-rossetti / intro-to-python

An Introduction to Programming in Python
Other
97 stars 244 forks source link

Category Encoders Package Docs #120

Open s2t2 opened 1 year ago

s2t2 commented 1 year ago

We need some notes about category encoders package:

Installation:

pip install category_encoders

Resources:

from category_encoders import OneHotEncoder, OrdinalEncoder

onehot = OneHotEncoder(handle_unknown="error", use_cat_names=True, return_df=True,
    cols=["categorical_column_1", "categorical_column_2"] # column names to one hot encode
)
encoded = onehot.fit_transform(df)
#print(onehot.feature_names_out_)
encoded.head()