matlab怎么使用随机森林算法进行分类训练
时间: 2024-02-06 18:02:30 浏览: 54
随机森林用于分类matlab代码
Matlab中可以使用`TreeBagger`函数来实现随机森林算法进行分类训练。下面是一个简单的示例代码:
```matlab
% 加载数据
load fisheriris
% 将数据拆分为训练集和测试集
idx = randperm(size(meas,1));
Xtrain = meas(idx(1:100),:);
Ytrain = species(idx(1:100));
Xtest = meas(idx(101:end),:);
Ytest = species(idx(101:end));
% 使用TreeBagger函数训练随机森林模型
ntrees = 50; % 随机森林中树的数量
model = TreeBagger(ntrees, Xtrain, Ytrain);
% 使用训练好的模型进行预测
Ypred = predict(model, Xtest);
% 计算分类准确率
accuracy = sum(strcmp(Ypred, Ytest))/length(Ytest);
fprintf('分类准确率为 %.2f%%\n', accuracy*100);
```
在上面的代码中,首先加载了一个经典的鸢尾花数据集。然后将数据集随机拆分为训练集和测试集。接着使用`TreeBagger`函数训练了一个包含50棵树的随机森林模型。最后使用训练好的模型对测试集进行预测,并计算分类准确率。
阅读全文