12311k

Untitled

Feb 16th, 2020
671
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.62 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.tree import DecisionTreeClassifier
  3. from sklearn.metrics import recall_score
  4. from sklearn.model_selection import train_test_split
  5.  
  6. data = pd.read_csv('/datasets/travel_insurance_preprocessed.csv')
  7.  
  8. target = data['Claim']
  9. features = data.drop('Claim', axis=1)
  10. features_train, features_valid, target_train, target_valid = train_test_split(
  11. features, target, test_size=0.25, random_state=12345)
  12.  
  13. model = DecisionTreeClassifier(random_state=12345)
  14. model.fit(features_train, target_train)
  15. predicted_valid = model.predict(features_valid)
  16.  
  17. print(recall_score(target_valid, predicted_valid))
Advertisement
Add Comment
Please, Sign In to add comment