使用matlab采用LightGBM实现回归预测的代码

以下是使用 MATLAB 实现 LightGBM 回归预测的示例代码：

% 加载数据
data = load('data.txt');
X = data(:, 1:end-1);
y = data(:, end);

% 将数据分为训练集和测试集
cv = cvpartition(size(X, 1), 'HoldOut', 0.3);
X_train = X(cv.training, :);
y_train = y(cv.training, :);
X_test = X(cv.test, :);
y_test = y(cv.test, :);

% 定义 LightGBM 参数
params = struct;
params.num_threads = 2;
params.objective = 'regression';
params.metric = 'mse';
params.learning_rate = 0.1;
params.num_leaves = 31;
params.min_data_in_leaf = 20;
params.feature_fraction = 0.8;
params.bagging_fraction = 0.8;
params.bagging_freq = 5;

% 训练 LightGBM 模型
lgb_train = lgb.Dataset(X_train, y_train);
lgb_test = lgb.Dataset(X_test, y_test, 'reference', lgb_train);
model = lgb.train(params, lgb_train, 100, lgb_test);

% 使用训练好的模型进行预测
y_pred = model.predict(X_test);
mse = mean((y_test - y_pred).^2);
fprintf('测试集 MSE：%f\n', mse);

在这个示例中，我们首先加载了数据并将其分成训练集和测试集。然后，我们定义了 LightGBM 的参数，包括使用的线程数、损失函数、评估指标、学习率、叶子节点数、叶子节点最小数据数、特征采样率等等。接下来，我们将训练集和测试集转换为 LightGBM 的数据集格式，并使用 `lgb.train` 函数训练模型。最后，我们使用训练好的模型在测试集上进行预测，并计算出 MSE 指标。