Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created December 18, 2020 14:57
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 amankharwal/1f6e62a87f365c5a8c89f3459f1908b6 to your computer and use it in GitHub Desktop.
Save amankharwal/1f6e62a87f365c5a8c89f3459f1908b6 to your computer and use it in GitHub Desktop.
from sklearn import preprocessing
def encode_features(df_train, df_test):
features = ['Fare', 'Cabin', 'Age', 'Sex', 'Lname', 'NamePrefix']
df_combined = pd.concat([df_train[features], df_test[features]])
for feature in features:
le = preprocessing.LabelEncoder()
le = le.fit(df_combined[feature])
df_train[feature] = le.transform(df_train[feature])
df_test[feature] = le.transform(df_test[feature])
return df_train, df_test
data_train, data_test = encode_features(data_train, data_test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment