目录
MATLAB实现XGBoost极限梯度提升树模型
以下是一个简单的示例代码:
% 准备数据
data = readtable('data.csv');
X = data(:,1:end-1);
y = data(:,end);
% 划分训练集和测试集
cv = cvpartition(size(X,1),'HoldOut',0.3);
idx_train = training(cv);
idx_test = test(cv);
% 定义模型参数
params.max_depth = 3;
params.eta = 0.1;
params.gamma = 1;
params.min_child_weight = 1;
params.subsample = 0.8;
params.colsa