在GLM.jl中仅指定目标变量

时间:2017-05-28 00:33:49

标签: metaprogramming julia

R中,可以仅使用公式对象指定目标变量:y ~ .

这不适用于GLM.jl

1 个答案:

答案 0 :(得分:2)

在R中,公式中的点符号表示除从属字段外的所有字段(在公式的LHS上)。显然GLM.jl没有实现这种表示法,但这可以改变。

例如,以下函数使事情更接近R:

using DataFrames    # install with Pkg.add("DataFrames")
using MacroTools    # install with Pkg.add("MacroTools")

function usedot!(f, df)
    vs = collect(filter(x->x != f.lhs,names(df)))
    ex = length(vs)>1 ? Expr(:call,:+,vs...) : vs[1]
    f.rhs = MacroTools.postwalk(x -> x == :(:.) ? ex : x , f.rhs)
    f.rhs = MacroTools.postwalk(x -> typeof(x)==Expr &&
      x.head==:quote && length(x.args)==1 &&
      x.args[1] == :. ? ex : x , f.rhs)
    return f
end

usedot(f,df) = usedot!(DataFrames.Formula(f),df)

这些功能可以按如下方式使用:

julia> using GLM, RDatasets

julia> form = dataset("datasets", "Formaldehyde");

julia> lm1 = fit(LinearModel, usedot(@formula(OptDen ~ :.),form), form)
DataFrames.DataFrameRegressionModel{GLM.LinearModel{GLM.LmResp{Array{Float64,1}},GLM.DensePredChol{Float64,Base.LinAlg.Cholesky{Float64,Array{Float64,2}}}},Array{Float64,2}}

Formula: OptDen ~ 1 + Carb

Coefficients:
               Estimate  Std.Error  t value Pr(>|t|)
(Intercept)  0.00508571 0.00783368 0.649211   0.5516
Carb           0.876286  0.0135345  64.7444    <1e-6

但是,当然,应该在实践中使用包含更多字段的DataFrame。 更改GLM.jl以包含点表示法也不应该这么难,但请注意在点具有具体含义之前需要将公式附加到DataFrame,这也许是迄今为止它被避免的原因。