Advertisement
Guest User

Untitled

a guest
Nov 17th, 2017
61
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.47 KB | None | 0 0
  1. # -*- coding: utf-8 -*-
  2. """
  3. Created on Sat Oct 28 19:28:05 2017
  4.  
  5. @author: Viet
  6. """
  7.  
  8. import pandas as pd
  9. from sklearn.ensemble import GradientBoostingClassifier
  10.  
  11. merged = pd.read_csv("merged.csv")  
  12. x_model = merged[["female","male"]].values
  13. y_model = merged["is_churn"].values
  14.  
  15. test = pd.read_csv("test_model.csv")
  16. x_test  = test[["female","male"]].values
  17. y_test = []
  18. clf = GradientBoostingClassifier().fit(X=x_model,y=y_model)
  19. y_test  = clf.predict_proba(x_test)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement