以下是使用matlab计算240个数据的随机森林特征重要性排序的代码:

  1. 加载数据
data = load('data.mat');
X = data.X;
Y = data.Y;
  1. 训练随机森林模型
numTrees = 100; % 设置树的数量
Mdl_rf = TreeBagger(numTrees, X, Y, 'Method', 'classification', 'OOBPrediction', 'on', 'OOBVarImp', 'on');
  1. 计算特征重要性
imp = Mdl_rf.OOBPermutedVarDeltaError; % 获取每个特征的重要性
[sorted_imp, idx] = sort(imp, 'descend'); % 对重要性进行排序
  1. 输出结果
for i = 1:length(idx)
    fprintf('Feature %d: %f\n', idx(i), sorted_imp(i));
end

输出结果将按降序排列的特征重要性显示


原文地址: https://www.cveoy.top/t/topic/diVz 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录