julia 中的多项式回归 - glm

Question

是否可以在 Julia 的 GLM 包中进行多项式回归？考虑到与 R 语法的相似性，我希望

 fit(LinearModel, @formula(y ~ poly(x,5)), dataset)

会起作用（用于拟合 5 次多项式）。它不是。

Answer 1

尽管您明确要求 GLM.jl 解决方案，但让我指出使用Polynomials.jl 的简单解决方案：

using Polynomials
polyfit(x, 5)

请参阅polyfit获取更多信息。

Answer 2

StatsModels.jl 不提供poly()函数，但他们的文档确实提供了一个完整的示例，展示了如何添加poly()将在宏内部工作的自己的函数@formula。