python – 使用scikit限制线性回归预测值

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

我正在考虑进行条件检查,如果预测超过10,我明确地将其设置为10.

有没有更好的办法？

解决方法:

如果y是回归对象的预测方法的输出,那么你可以将Numpy的最小值限制为10：

y = np.minimum(y, 10.)

要把它限制在零以下,做

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

或者,更短：

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

标签：python,linear-regression,scikits
来源： https://codeday.me/bug/20190826/1730230.html