Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@amandaiglesiasmoreno

amandaiglesiasmoreno/label_encoding.py Secret

Last active November 22, 2021 21:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amandaiglesiasmoreno/2f451cc0a6966e997b936e9a3c49352a to your computer and use it in GitHub Desktop.
Save amandaiglesiasmoreno/2f451cc0a6966e997b936e9a3c49352a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
label_encoding.py
df_telco_transformed = df_telco.copy()
# label encoding (binary variables)
label_encoding_columns = ['gender', 'Partner', 'Dependents', 'PaperlessBilling', 'PhoneService', 'Churn']
# encode categorical binary features using label encoding
for column in label_encoding_columns:
if column == 'gender':
df_telco_transformed[column] = df_telco_transformed[column].map({'Female': 1, 'Male': 0})
else:
df_telco_transformed[column] = df_telco_transformed[column].map({'Yes': 1, 'No': 0})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment