Guest User

Untitled

a guest
Mar 4th, 2016
339
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.47 KB | None | 0 0
  1. import pandas as pd
  2. import numpy as np
  3. from sklearn.tree import DecisionTreeClassifier
  4.  
  5. data = pd.read_csv('titanic.csv', index_col = 'PassengerId')
  6.  
  7. data = data.dropna()
  8. data['Sex'] = np.where(data['Sex'] == 'male', 1, 0)
  9.  
  10. features = data[['Pclass', 'Fare', 'Age', 'Sex']]
  11. target = data['Survived']
  12.  
  13. clf = DecisionTreeClassifier(random_state = 241)
  14. clf.fit(features, target)
  15.  
  16. clf.feature_importances_
  17. Out: array([ 0.01793336,  0.29549222,  0.40310552,  0.2834689 ])
Advertisement
Add Comment
Please, Sign In to add comment