机器学习 集成学习方法之随机森林

发布于:2025-05-21 ⋅ 阅读:(22) ⋅ 点赞:(0)

1 集成学习

机器学习中有一种大类叫集成学习(Ensemble Learning),集成学习的基本思想就是将多个分类器组合,从而实现一个预测效果更好的集成分类器。集成算法可以说从一方面验证了中国的一句老话:三个臭皮匠,赛过诸葛亮。集成算法大致可以分为:Bagging,Boosting 和 Stacking 三大类型。

(1)每次有放回地从训练集中取出 n 个训练样本,组成新的训练集;

(2)利用新的训练集,训练得到M个子模型;

(3)对于分类问题,采用投票的方法,得票最多子模型的分类类别为最终的类别;

2 随机森林的算法原理

随机森林就属于集成学习,是通过构建一个包含多个决策树(通常称为基学习器或弱学习器)的森林, 每棵树都在不同的数据子集和特征子集上进行训练,最终通过投票或平均预测结果来产生更准确和稳健的预测。这种方法不仅提高了预测精度,也降低了过拟合风险,并且能够处理高维度和大规模数据集

在这里插入图片描述
特点:

  • 随机: 特征随机,训练集随机
    • 样本:对于一个总体训练集T,T中共有N个样本,每次有放回地随机选择n个样本。用这n个样本来训练一个决策树。
    • 特征:假设训练集的特征个数为d,每次仅选择k(k<d)个来构建决策树。
  • 森林: 多个决策树分类器构成的分类器, 因为随机,所以可以生成多个决策树
  • 处理具有高维特征的输入样本,而且不需要降维
  • 使用平均或者投票来提高预测精度和控制过拟合
    在这里插入图片描述

2.1 Sklearn API

class sklearn.ensemble.RandomForestClassifier

参数:
n_estimators	int, default=100
森林中树木的数量。(决策树个数)

criterion	{“gini”, “entropy”}, default=”gini” 决策树属性划分算法选择
	当criterion取值为“gini”时采用 基尼不纯度(Gini impurity)算法构造决策树,
	当criterion取值为 “entropy” 时采用信息增益( information gain)算法构造决策树.
	
max_depth	int, default=None 树的最大深度。 

2.2 示例

坦尼克号乘客生存
代码如下:

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.feature_extraction import DictVectorizer
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import GridSearchCV
import joblib
def train():
    # 数据集加载
    titanic=pd.read_csv(r"..\22day4.25机器学习\src\titanic\titanic.csv")
    # 数据集处理
    #获取关键特征
    titanic=titanic[['age','pclass','sex','survived']]
    # 将其中的缺省值赋值为这个列的平均值
    titanic["age"].fillna(titanic["age"].mean(),inplace=True)
    # 获取特征值和目标值
    x=titanic[['age','pclass','sex']]
    y=titanic[['survived']].to_numpy()
    # 将x转化为字典
    x=x.to_dict(orient='records')
    # 字典向量化
    vac=DictVectorizer(sparse=True)
    x=vac.fit_transform(x).toarray()
    # 划分数据集
    x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.2,random_state=22,shuffle=True)
    # 标准化
    transfer=StandardScaler()
    x_train=transfer.fit_transform(x_train)
    x_test=transfer.transform(x_train)
    print(x_test.shape,y_test.shape)
    # 模型建立
    model=RandomForestClassifier(n_estimators=10,max_depth=4)
    # 训练模型
    model.fit(x_train,y_train)
    # 模型评估
    score=model.score(x_test,y_test)
    print(score)
    # 保存模型
    joblib.dump(model,r"..\23day5.8\src\model\rf.pkl")
    joblib.dump(transfer,r"..\23day5.8\src\model\rf_transfer.pkl")
    joblib.dump(vac,r"..\23day5.8\src\model\rf_vac.pkl")
def detect():
    model=joblib.load(r"..\23day5.8\src\model\rf.pkl")
    transfer=joblib.load(r"..\23day5.8\src\model\rf_transfer.pkl")
    vac=joblib.load(r"..\23day5.8\src\model\rf_vac.pkl")
    x_test=[{'age':24,'pclass':'1st','sex':"male"}]
    x_test=vac.transform(x_test).toarray()
    # print(x_test)
    x_test=transfer.transform(x_test)
    prd=model.predict(x_test)
    print(prd)
if __name__=="__main__":
    train()
    # predict()
    detect()


网站公告

今日签到

点亮在社区的每一天
去签到