一周算法实践---金融贷款逾期模型

金融贷款逾期模型

1.读取数据

import pandas as pd
data_all = pd.read_csv('../data/data_all.csv')`

2.划分数据集

from sklearn.model_selection import train_test_split
x = data_all.drop(['status'], axis=1)
y = data_all['status']
#数据三七分,随机种子2018
x_train, x_test, y_train, y_test = train_test_split(x, y,     
test_size=0.3,random_state=2018)

3.模型训练

#搭建模型
from sklearn.tree import DecisionTreeClassifier
from sklearn import svm
from sklearn.linear_model import LogisticRegression
#逻辑回归模型
lr = LogisticRegression(random_state =2018)
lr.fit(x_train, y_train)
#支持向量机
svc = svm.SVC(random_state =2018)
svc.fit(x_train,y_train)
#决策树
clf = DecisionTreeClassifier(random_state =2018)
clf.fit(x_train,y_train)

4.模型评估

   #评分
   lr_score = lr.score(X_test,y_test)
   Svc_score = Svc.score(X_test,y_test)
   dt_score = dt.score(X_test,y_test
   print(lr_score)
   print(Svc_score)
   print(dt_score)
   #得到结果
   (0.7484232655921513, 0.7484232655921513, 0.6846531184302733)