百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程字典 > 正文

机器学习之决策树算法,如何选择相亲对象

toyiye 2024-07-03 02:04 20 浏览 0 评论

什么是决策树算法?

决策树本质是一种树结构,预测时,在树的内部节点处用某一属性值进行判断,根据判断结果决定进入哪个分支节点,直到到达叶节点处,得到分类结果。本质上就是不断进行 if ... else ... 判断。

比如,一位女士,28 岁,未婚,过年回家被妈妈介绍相亲。

对话1

妈妈:给你介绍一位男士。

女儿:长相如何?

妈妈:耐看型,不算丑。

女儿:性格怎么样?

妈妈:不好。

女儿:不见。

对话2

妈妈:给你介绍一位男士。

女儿:长相如何?

妈妈:耐看型,不算丑。

女儿:收入如何?

妈妈:very OK.

女儿:见。

以上决策的过程就是决策树的原理。

决策依据包括长相、收入、性格、职位。决策结果为见或不见。

当这种对话多进行几轮后,我们就可以大致判断出哪些因素对结果影响较大,比如收入高或者职位是公务员更加受青睐,而其他因素影响甚微。那么,程序如何选择条件判断的先后次序呢?

决策树分类原理:熵

系统越有序,熵值越低;系统越混乱或者分散,熵值越高

在信息论中,熵是表示随机变量Y的不确定性的度量。假设类别Y是一个离散随机变量,其概率分布为p1,p2,...,pK,那么Y的熵被定义为:

一般,对数取以2为底。

例如:

足球比赛里,有4个球队 {A,B,C,D} ,获胜概率分别为{1/2, 1/4, 1/8, 1/8},求H(X)。

信息增益

信息增益:以某特征划分数据集前后的熵的差值。熵可以表示样本集合的不确定性,熵越大,样本的不确定性就越大。因此可以使用划分前后集合熵的差值来衡量使用当前特征对于样本集合D划分效果的好坏

G(D|A)=H(D)-H(D|A)

信息熵:

条件熵:

Ck表示某个类别的样本数。

例如:

第一列为论坛号码,第二列为性别,第三列为活跃度,最后一列用户是否流失。

那么性别和活跃度两个特征,哪个对用户流失影响更大

整体熵:

性别熵:

性别信息增益:

活跃度熵:

活跃度信息增益:

Python程序实战

预测波士顿房屋价格

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import sklearn.model_selection as ms # 模型选择
import sklearn.svm as svm
import sklearn.metrics as sm
import sklearn.tree as st #决策树

# 读取数据
data = pd.read_csv('./HousingData.csv')
y = data['MEDV']
x = data.drop('MEDV', axis=1)

train_x, \
test_x, \
train_y, \
test_y = ms.train_test_split(x, y,#待划分数据
                             test_size=0.2,#测试集占比
                             random_state=7)#随机种子


#单棵决策树回归
model = st.DecisionTreeRegressor(max_depth=4)
model.fit(train_x,train_y)
pred_train_y = model.predict(train_x)
pred_test_y = model.predict(test_x)
print('单棵数训练集:',sm.r2_score(train_y,pred_train_y))
print('单棵数测试集:',sm.r2_score(test_y,pred_test_y))

#特征重要性
fi = model.feature_importances_
print(fi)
# print(fi.sum())
fi = pd.Series(fi)

fi = fi.sort_values(ascending=False)
print(fi)

#决策树可视化
st.plot_tree(model)

plt.show()

集成学习

梯度提升决策树(GBDT Gradient Boosting Decision Tree) 是一种迭代的决策树算法该算法由多棵决策树组成,所有树的结论累加起来做最终答案。它在被提出之初就被认为是泛化能力(generalization)较强的算法。

# 删除缺失值
data = data.dropna()

#构建模型
sub_model = st.DecisionTreeRegressor(max_depth=6)
model = se.AdaBoostRegressor(sub_model,
                             n_estimators=400,
                             random_state=7)
model.fit(train_x,train_y)
pred_train_y = model.predict(train_x)
pred_test_y = model.predict(test_x)
print('训练集:',sm.r2_score(train_y,pred_train_y))
print('测试集:',sm.r2_score(test_y,pred_test_y))

输出:
训练集: 0.985593080950688
测试集: 0.9264863823619695


相关推荐

为何越来越多的编程语言使用JSON(为什么编程)

JSON是JavascriptObjectNotation的缩写,意思是Javascript对象表示法,是一种易于人类阅读和对编程友好的文本数据传递方法,是JavaScript语言规范定义的一个子...

何时在数据库中使用 JSON(数据库用json格式存储)

在本文中,您将了解何时应考虑将JSON数据类型添加到表中以及何时应避免使用它们。每天?分享?最新?软件?开发?,Devops,敏捷?,测试?以及?项目?管理?最新?,最热门?的?文章?,每天?花?...

MySQL 从零开始:05 数据类型(mysql数据类型有哪些,并举例)

前面的讲解中已经接触到了表的创建,表的创建是对字段的声明,比如:上述语句声明了字段的名称、类型、所占空间、默认值和是否可以为空等信息。其中的int、varchar、char和decimal都...

JSON对象花样进阶(json格式对象)

一、引言在现代Web开发中,JSON(JavaScriptObjectNotation)已经成为数据交换的标准格式。无论是从前端向后端发送数据,还是从后端接收数据,JSON都是不可或缺的一部分。...

深入理解 JSON 和 Form-data(json和formdata提交区别)

在讨论现代网络开发与API设计的语境下,理解客户端和服务器间如何有效且可靠地交换数据变得尤为关键。这里,特别值得关注的是两种主流数据格式:...

JSON 语法(json 语法 priority)

JSON语法是JavaScript语法的子集。JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔花括号保存对象方括号保存数组JS...

JSON语法详解(json的语法规则)

JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔大括号保存对象中括号保存数组注意:json的key是字符串,且必须是双引号,不能是单引号...

MySQL JSON数据类型操作(mysql的json)

概述mysql自5.7.8版本开始,就支持了json结构的数据存储和查询,这表明了mysql也在不断的学习和增加nosql数据库的有点。但mysql毕竟是关系型数据库,在处理json这种非结构化的数据...

JSON的数据模式(json数据格式示例)

像XML模式一样,JSON数据格式也有Schema,这是一个基于JSON格式的规范。JSON模式也以JSON格式编写。它用于验证JSON数据。JSON模式示例以下代码显示了基本的JSON模式。{"...

前端学习——JSON格式详解(后端json格式)

JSON(JavaScriptObjectNotation)是一种轻量级的数据交换格式。易于人阅读和编写。同时也易于机器解析和生成。它基于JavaScriptProgrammingLa...

什么是 JSON:详解 JSON 及其优势(什么叫json)

现在程序员还有谁不知道JSON吗?无论对于前端还是后端,JSON都是一种常见的数据格式。那么JSON到底是什么呢?JSON的定义...

PostgreSQL JSON 类型:处理结构化数据

PostgreSQL提供JSON类型,以存储结构化数据。JSON是一种开放的数据格式,可用于存储各种类型的值。什么是JSON类型?JSON类型表示JSON(JavaScriptO...

JavaScript:JSON、三种包装类(javascript 包)

JOSN:我们希望可以将一个对象在不同的语言中进行传递,以达到通信的目的,最佳方式就是将一个对象转换为字符串的形式JSON(JavaScriptObjectNotation)-JS的对象表示法...

Python数据分析 只要1分钟 教你玩转JSON 全程干货

Json简介:Json,全名JavaScriptObjectNotation,JSON(JavaScriptObjectNotation(记号、标记))是一种轻量级的数据交换格式。它基于J...

比较一下JSON与XML两种数据格式?(json和xml哪个好)

JSON(JavaScriptObjectNotation)和XML(eXtensibleMarkupLanguage)是在日常开发中比较常用的两种数据格式,它们主要的作用就是用来进行数据的传...

取消回复欢迎 发表评论:

请填写验证码