各位同学好,今天我和大家分享一下python机器学习的随机森林算法。主要内容有:

(1) 算法原理、(2) sklearn实现随机森林、(3) 泰坦尼克号沉船幸存者预测。

文末附python完整代码和数据集。那我们开始吧。


1. 随机森林算法原理

集成算法包含(bagging装袋/boosting增强/stacking堆叠),其中随机森林属于bagging。

算法原理:

        随机森林是一个包含多个决策树的分类器,并且其输出的类别是由个别树输出的类别的众数而定。利用相同的训练数搭建多个独立的分类模型,然后通过投票的方法,以少数服从多数原则作出最终的分类决策。

例如:训练了5个树,其中4个树的结果是True,1个树的结果是False,最终结果也会是True。

学习算法:

若用N表示训练用的样本个数,M表示特征数目。输入特征数目m,用于确定决策树上一个节点的决策结果,其中m远小于M。

从N个训练样本中以有放回抽样的方法,取样M次,形成一个训练集,并用未抽到的样本做预测,评估其误差。

对于每一个节点,随机选择m个特征,决策树上每个节点的决定都是基于这些特征决定的。根据这m个特征,计算其最佳的分裂方式。

随机抽样训练集的原因:

如果不进行随机抽样,每棵树的训练集都一样,那么最终训练出的树分类结果也是完全一样的。

为什么有放回地抽样:

如果不进行有放回抽样,那么每棵树的训练集都是完全不相同的,都是没有交集的,每棵树训练出来都是有很大差异,而随机森林最后分类取决于多棵树的投票。


2. 随机森林算法实战

2.1 sklearn实现随机森林

首先从sklearn中导入随机森林分类器:

from sklearn.ensemble import RandomForestClassifier  

随机森林函数 RandomForestClassifier()  参数设置:

n_estimator:整数类型,森林里树木的数量,默认=10

criterion:字符串型,分割特征的测量方法,默认是'Gini',可选信息熵 'entropy'

max_depth:整数或None,树的最大深度,默认是None

bootstrap:布尔,是否采用有放回抽样,默认是True


2.2 实例应用

随机森林算法和决策树算法在程序上有类似部分,该篇文章就不再赘述,如果以下程序语句有不明白的,可以参考上一篇文章,使用的数据集也相同:【机器学习】(5) 决策树算法实战:sklearn实现决策树,实例应用(沉船幸存者预测)附python完整代码及数据集

(1)导入数据

        导入泰坦尼克号乘客数据,数据集获取地址:GitHub - fayduan/Kaggle_Titanic: the data and ipython notebook of my attempt to solve the kaggle titanic problem,数据包含891个样本,10项特征数据姓名、年龄、船舱等,一项目标数据'survived'。

随机森林算法:原理、实例应用(沉船幸存者预测)附python完整代码和数据集_python

(2)数据预处理

        从原数据中提取五项指标作为特征值,便于大家对随机森林算法有更直观的理解。由于暂时文章没有涉及到特征工程,暂且不把所有的特征值放入模型,后续文章会进行跟深入的探讨。

        使用年龄的平均值并借助.fillna()函数填充age特征中的缺失值,划分出建模所需的数据验证模型所需的数据

(3)特征提取

        由于特征sex对应的数据是'female''male',属于字符串类型数据,然而模型训练函数.fit()只能接收数值类型的数据,因此需要将特征值转换成sparse数值矩阵。之前我们提到过文本特征提取方法CountVectorizer(),这里我们使用字典特征提取方法DictVectorizer(),要求输入的参数是一个由字典组成的列表,而现在的特征值是一个DataFrame类型数据;返回值是sparse数值矩阵,将字符串类型数据转变成数值类型。

使用 .to_dict(orient = 'record') 方法将DataFrame类型数据转变成由字典组成的列表

有关sparse矩阵文本特征抽取的知识见下文的第1.4小节:

【机器学习】(2) 朴素贝叶斯算法:原理、实例应用(文档分类预测)附python完整代码及数据集

sparse矩阵如下图所示:索引3对应female,索引4对应male,将性别转换成数值类型,索引3表示,是否是'female',是就显示1,不是显示0。

随机森林算法:原理、实例应用(沉船幸存者预测)附python完整代码和数据集_python_02

(4)划分训练集和测试集

         一般采用75%的数据用于训练,25%用于测试,因此把数据进行训练之前,先要对数据划分。划分方法不再赘述,有疑惑的可看下文中的第2.3节:【机器学习】(2) 朴素贝叶斯算法:原理、实例应用(文档分类预测)附python完整代码及数据集

(5) 随机森林方法

变量 rf 接收随机森林分类器 RandomForestClassifier()随机森林分类器的参数同决策树的参数,这里就不赘述,具体见下文的第1节:

【机器学习】(5) 决策树算法实战:sklearn实现决策树,实例应用(沉船幸存者预测)附python完整代码及数据集

训练函数 .fit() 中传入训练所用的x和y值,其中x_train是sparse数值矩阵,accuracy存放评分法 .score() 求得的模型误差,根据x_test预测结果,把结果和真实的y_test比较,计算准确率。result中存放预测函数 .predict() 得到的人员存活结果,预测函数的输入值data_predict_features也是sparse数值类型矩阵。

最终模型准确率在80%左右,预测结果和实际结果见稍有偏差。

随机森林算法:原理、实例应用(沉船幸存者预测)附python完整代码和数据集_数据_03


数据集获取:

GitHub - fayduan/Kaggle_Titanic: the data and ipython notebook of my attempt to solve the kaggle titanic problem

完整python代码: