岭回归matlab代码
时间: 2023-11-25 09:30:28 浏览: 52
岭回归代码
以下是一个简单的岭回归示例的 MATLAB 代码:
% 生成一些随机数据
X = randn(100, 10); % 这里使用了10个特征
y = randn(100, 1);
% 划分训练集和测试集
train_size = 70;
X_train = X(1:train_size, :);
y_train = y(1:train_size);
X_test = X(train_size+1:end, :);
y_test = y(train_size+1:end);
% 使用岭回归拟合模型
lambda = 0.1; % 正则化参数
beta_ridge = ridge(y_train, X_train, lambda);
% 计算在测试集上的预测误差
y_pred = X_test * beta_ridge(2:end) + beta_ridge(1); % 注意要去掉截距项
mse = mean((y_pred - y_test).^2);
% 输出结果
fprintf('岭回归模型的测试集均方误差为:%.4f\n', mse);
注:这里使用了 MATLAB 自带的 ridge 函数进行岭回归。
阅读全文