Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@escuccim

escuccim/mean_encode.py

Created June 7, 2018 11:21
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 escuccim/cbb2c0140ae15fba7184e8c3918f9160 to your computer and use it in GitHub Desktop.
Save escuccim/cbb2c0140ae15fba7184e8c3918f9160 to your computer and use it in GitHub Desktop.
Download ZIP
mean encoding in python
Raw
mean_encode.py
# mean encoding
means = X_tr.groupby(col).target.mean()
train_new[col+'_target_mean'] = train_new[col].map(means)
val_new[col+'_target_mean'] = val_new[col].map(means)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment