随机森林模型MATLAB
时间: 2023-06-30 19:09:08 浏览: 166
MATLAB 随机森林模型
MATLAB可以使用TreeBagger函数来实现随机森林模型。下面是一个简单的示例:
1. 准备数据,将数据集划分为训练集和测试集。
2. 使用TreeBagger函数创建随机森林模型。可以设置树的数量、特征数量等参数。
3. 使用训练集训练模型。
4. 使用测试集测试模型,并计算模型的准确率、召回率等指标。
下面是一个具体的MATLAB代码示例:
```matlab
% 准备数据
load fisheriris
X = meas;
Y = species;
rng(1); % 设置随机种子
cv = cvpartition(Y,'HoldOut',0.3); % 将数据集划分为训练集和测试集
idxTrain = training(cv);
idxTest = test(cv);
XTrain = X(idxTrain,:);
YTrain = Y(idxTrain);
XTest = X(idxTest,:);
YTest = Y(idxTest);
% 创建随机森林模型
nTrees = 50;
mdl = TreeBagger(nTrees,XTrain,YTrain,'OOBPrediction','On','Method','classification');
% 训练模型
mdl = mdl.fit(XTrain,YTrain);
% 测试模型
YTest_pred = predict(mdl,XTest);
confMat = confusionmat(YTest,YTest_pred); % 计算混淆矩阵
accuracy = sum(diag(confMat))/sum(confMat(:)); % 计算准确率
precision = diag(confMat)./sum(confMat,2); % 计算精确率
recall = diag(confMat)./sum(confMat,1)'; % 计算召回率
```
在上面的示例中,我们使用了iris数据集,将数据集划分为训练集和测试集,创建了一个包含50棵树的随机森林模型,并使用训练集训练了模型。最后,我们使用测试集测试了模型,并计算了模型的准确率、精确率和召回率。
阅读全文