给我一个整体的随机森林回归matlab代码列子
时间: 2023-12-12 11:04:09 浏览: 95
以下是一个完整的随机森林回归MATLAB代码示例,用于预测波士顿房价。代码中使用了波士顿房价数据集,包含506个样本和13个特征。其中,前12个特征用于预测房价,最后一个特征是目标变量(即房价)。
```matlab
% 加载数据
load('boston.mat');
X = boston(:,1:12);
Y = boston(:,13);
% 将数据集拆分为训练集和测试集
cv = cvpartition(size(X,1),'HoldOut',0.3);
idx = cv.test;
Xtrain = X(~idx,:);
Ytrain = Y(~idx,:);
Xtest = X(idx,:);
Ytest = Y(idx,:);
% 训练随机森林模型
nTrees = 100;
B = TreeBagger(nTrees,Xtrain,Ytrain,'Method','regression');
% 预测测试集数据
Yfit = predict(B,Xtest);
% 计算预测结果的误差
RMSE = sqrt(mean((Ytest-str2double(Yfit)).^2));
% 输出结果
fprintf('随机森林回归模型的均方根误差(RMSE)为:%.2f\n', RMSE);
```
在这个例子中,我们使用了`TreeBagger`函数训练了一个包含100棵树的随机森林回归模型,然后使用`predict`函数预测了测试集数据,并计算了预测结果的均方根误差。
阅读全文