随机森林-sklearn.ensemble.RandomForestRegressor

本文介绍了如何使用sklearn.ensemble模块的RandomForestRegressor进行随机森林回归。该模型通过构建多个决策树并取平均值来提高预测准确性和防止过拟合。示例代码展示了训练和预测过程。

随机森林回归

class sklearn.ensemble.RandomForestRegressor(n_estimators=10criterion=’mse’max_depth=None,min_samples_split=2min_samples_leaf=1min_weight_fraction_leaf=0.0max_features=’auto’max_leaf_nodes=None,min_impurity_decrease=0.0min_impurity_split=Nonebootstrap=Trueoob_score=Falsen_jobs=1random_state=None,verbose=0warm_start=False)

随机森林是一种目标估计,通过对数据集上的部分样本形成一个分类决策树,并使用averaging去提高预测准确率和控制过拟合发生。

注:http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html


用法:

>>>from sklearn.ensemble import RandomForestRegressor

>>>data=[[0,0,0],[1,1,1],[2,2,2]]

>>>target=[0,1,2]

>>>rfr=RandomForestRegressor()

>>>rfr.fit(data,target)   #训练数据

>>>print(rfr.predict([[1,1,1]]))    #预测数据

[1.]

>>>print(rfr.predict([[1,1,1],[2,2,2]]))

[0.7  1.8]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值