Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- data['Cabin_mapped'] = data['Cabin'].astype(str).str[0]
- # this transforms the letters into numbers
- cabin_dict = {k:i for i, k in enumerate(
- data['Cabin_mapped'].unique(),0)}
- data.loc[:,'Cabin_mapped'] = data.loc[:,'Cabin_mapped'].map(cabin_dict)
- data[['Cabin_mapped', 'Cabin']].head()
- X_train_less_cat, X_test_less_cat, y_train, y_test =
- train_test_split(data[use_cols].fillna(0), data.Survived,
- test_size = 0.3, random_state=0)
- rf = RandomForestClassifier(n_estimators=200, random_state=39)
- rf.fit(X_train_less_cat, y_train)
Add Comment
Please, Sign In to add comment