在R中,可以仅使用公式对象指定目标变量:y ~ .
这不适用于GLM.jl
。
答案 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,这也许是迄今为止它被避免的原因。