使用计算240个数据的matlab随机森林特征重要性排序代码
以下是使用matlab计算240个数据的随机森林特征重要性排序的代码:
- 加载数据
data = load('data.mat');
X = data.X;
Y = data.Y;
- 训练随机森林模型
numTrees = 100; % 设置树的数量
Mdl_rf = TreeBagger(numTrees, X, Y, 'Method', 'classification', 'OOBPrediction', 'on', 'OOBVarImp', 'on');
- 计算特征重要性
imp = Mdl_rf.OOBPermutedVarDeltaError; % 获取每个特征的重要性
[sorted_imp, idx] = sort(imp, 'descend'); % 对重要性进行排序
- 输出结果
for i = 1:length(idx)
fprintf('Feature %d: %f\n', idx(i), sorted_imp(i));
end
输出结果将按降序排列的特征重要性显示
原文地址: https://www.cveoy.top/t/topic/diVz 著作权归作者所有。请勿转载和采集!