数学建模学习(2)——决策树

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score
df=pd.read_excel('股票客户流失.xlsx')
x=df.drop(columns='是否流失')#x等于除'是否流失'这一列以外的数据
y=df['是否流失']
#test_size的值代表测试集的比例是2,训练集的比例是8;random_state的值让结果统一
x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.2,random_state=1)
model=DecisionTreeClassifier(max_depth=5,random_state=1)#max_depth控制决策树的最大深度
model.fit(x_train,y_train)#训练集训练模型
y_pred=model.predict(x_test)#利用训练后的模型预测
score=accuracy_score(y_pred,y_test)#比较预测准确度
print(y_pred[0:100])  # 打印预测内容的前100个看看
print(score)

部分数据集如下:

结果如下:

相关推荐

  1. 机器学习-决策

    2024-07-23 09:40:04       42 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-23 09:40:04       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-23 09:40:04       102 阅读
  3. 在Django里面运行非项目文件

    2024-07-23 09:40:04       83 阅读
  4. Python语言-面向对象

    2024-07-23 09:40:04       92 阅读

热门阅读

  1. uniapp刷新当前页面bug

    2024-07-23 09:40:04       24 阅读
  2. ArcGIS Pro SDK (九)几何 12 多面体

    2024-07-23 09:40:04       21 阅读
  3. 数据库连接池

    2024-07-23 09:40:04       25 阅读
  4. RKNN执行bash ./build-linux_RK3566_RK3568.sh 报错

    2024-07-23 09:40:04       22 阅读
  5. 数学建模--图论与最短路径

    2024-07-23 09:40:04       21 阅读
  6. mariadb安装在服务器(Linux)

    2024-07-23 09:40:04       24 阅读
  7. 银行卡二三四要素核验的多种应用场景

    2024-07-23 09:40:04       28 阅读
  8. 怎么在 Ubuntu 中卸载已经安装的软件

    2024-07-23 09:40:04       23 阅读
  9. ubuntu 源码安装postgis插件

    2024-07-23 09:40:04       28 阅读
  10. SpringCloud-Zuul-03

    2024-07-23 09:40:04       27 阅读