用scikit封顶线性回归预测值

Question

我正在使用一个实值标签在区间0,10的数据集训练线性回归模型。我在测试集中的预测值有一些超过10的预测值。有没有方法将预测值限制在10。

我正在考虑做一个条件检查，这样如果预测超过10，我就显式地将它设置为10。

有没有更好的方法？

Answer 1

如果y是回归对象的predict方法的输出，那么您可以使用Numpy的minimum将其限制为10：

y = np.minimum(y, 10.)

要将其限制在零以下，请执行以下操作

y = np.maximum(np.minimum(y, 10.), 0.)

或者，简称：

y = np.clip(y, 0., 10.)