matlab决策树代码
时间: 2023-08-12 18:08:14 浏览: 781
基于Matlab实现决策树与随机森林算法(源码+数据+说明文档).rar
5星 · 资源好评率100%
以下是一个使用MATLAB编写的简单决策树代码示例:
```matlab
% 加载示例数据集(这里使用鸢尾花数据集)
load fisheriris
X = meas;
y = species;
% 将数据集拆分为训练集和测试集
rng(42); % 设置随机种子以确保结果可重现
cv = cvpartition(y, 'HoldOut', 0.2);
X_train = X(training(cv), :);
y_train = y(training(cv), :);
X_test = X(test(cv), :);
y_test = y(test(cv), :);
% 创建决策树模型
tree = fitctree(X_train, y_train);
% 在测试集上进行预测
y_pred = predict(tree, X_test);
% 计算准确率
accuracy = sum(strcmp(y_pred, y_test)) / numel(y_test);
disp(['准确率:', num2str(accuracy)]);
```
这段代码使用了MATLAB的`fitctree`函数来构建决策树模型,并使用鸢尾花数据集进行训练和测试。最后输出了模型在测试集上的准确率。
请注意,这只是一个简单的示例,实际应用中可能需要更多的预处理和调参步骤来优化模型性能。
阅读全文