如何从sklearn LinearRegression中导出线性回归公式

时间:2015-11-16 10:14:33

标签: python scikit-learn linear-regression

我想拥有模型的公式,以便在其他语言/项目中使用它。有没有办法从模型中导出公式?

我将使用sklearn linear regression model

我最终想做什么:给定公式f()和数据集'd',我将有java脚本代码,它将根据f()给出d的预测。

1 个答案:

答案 0 :(得分:4)

该公式可以基本上由学习的系数来描述。可以使用属性coef_intercept_获得系数。系数和输入矢量之间的点积加上截距给出了模型的输出。

在scikit-learn中实现此“公式”的实际代码类似于:

return safe_sparse_dot(X, self.coef_.T,
                       dense_output=True) + self.intercept_

对于您来说,移植到您的其他项目应该不会太困难。