Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """
- Created on Sat Oct 28 19:28:05 2017
- @author: Viet
- """
- import pandas as pd
- from sklearn.ensemble import GradientBoostingClassifier
- merged = pd.read_csv("merged.csv")
- x_model = merged[["female","male"]].values
- y_model = merged["is_churn"].values
- test = pd.read_csv("test_model.csv")
- x_test = test[["female","male"]].values
- y_test = []
- clf = GradientBoostingClassifier().fit(X=x_model,y=y_model)
- y_test = clf.predict_proba(x_test)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement