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

Question

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

我将使用 sklearn linear regression model .

我最终想做的是:给定一个公式 f() 和数据集“d”，我将拥有 java 脚本代码，它会根据 f() 对 d 进行预测。

Answer 1

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

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

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

移植到您的其他项目应该不会太困难。