运行线性Kernel Ridge回归拟合时报维度错误该如何解决

错误原因 #

• 问题出现在代码的最后一行预测调用：你调用KernelRidge的predict()方法时，错误传入了y_test作为入参
• predict()方法要求输入的是和训练集特征维度匹配的2D特征数组，而y_test是拆分得到的1维真实标签数组，自然触发维度不匹配的报错。

修正方法 #

将最后一行代码：

y_pred_kr = kr.predict(y_test)

替换为：

y_pred_kr = kr.predict(x_test)

即可正常完成预测。如果需要评估模型效果，可后续用输出的y_pred_kr和y_test计算回归评价指标。

内容的提问来源于stack exchange，提问作者Stunts1897