Guest User

Untitled

a guest
Sep 20th, 2018
83
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.56 KB | None | 0 0
  1. data['Cabin_mapped'] = data['Cabin'].astype(str).str[0]
  2. # this transforms the letters into numbers
  3. cabin_dict = {k:i for i, k in enumerate(
  4. data['Cabin_mapped'].unique(),0)}
  5.  
  6. data.loc[:,'Cabin_mapped'] = data.loc[:,'Cabin_mapped'].map(cabin_dict)
  7.  
  8. data[['Cabin_mapped', 'Cabin']].head()
  9.  
  10.  
  11.  
  12. X_train_less_cat, X_test_less_cat, y_train, y_test =
  13. train_test_split(data[use_cols].fillna(0), data.Survived,
  14. test_size = 0.3, random_state=0)
  15. rf = RandomForestClassifier(n_estimators=200, random_state=39)
  16. rf.fit(X_train_less_cat, y_train)
Add Comment
Please, Sign In to add comment