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

Question

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

我将使用sklearn 线性回归模型。

我最终想要做的是：给定一个公式 f() 和数据集“d”，我将拥有 Java 脚本代码，该代码将基于 f() 为我提供对 d 的预测。

Answer 1

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

在 scikit-learn 中实现这个“公式”的实际代码是这样的：

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

这对您来说移植到其他项目应该不会太难。