使用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.)