Julia：为子类型创建方法（LinearModel）

Question

我正在尝试为 GLM 包的 LinearModel 子类型创建一个方法。不幸的是，这似乎不起作用：

using DataFrames, GLM, RDatasets

iris = dataset("datasets", "iris")

fit = lm(@formula(SepalLength ~ PetalWidth), iris)

function print_r2(fit::GLM.LinearModel)

    R2 = r2(fit)
    println("The model's R2 is $R2.")

end

print_r2(fit)

还有错误：

MethodError: no method matching print_r2(::StatsModels.DataFrameRegressionModel{GLM.LinearModel{GLM.LmResp{Array{Float64,1}},GLM.DensePredChol{Float64,Base.LinAlg.Cholesky{Float64,Array{Float64,2}}}},Array{Float64,2}})
Closest candidates are:
  print_r2(::GLM.LinearModel) at In[1]:9

我必须指定整个类型吗？

Answer 1

你应该指定类型有点不同：

import StatsBase

function print_r2(fit::StatsBase.RegressionModel)
    R2 = r2(fit)
    println("The model's R2 is $R2.")
end

或fit::StatsModels.DataFrameRegressionModel，如果您想更具体一点。

问题是你的fit 不是GLM.LinearModel 类型：

julia> fit isa GLM.LinearModel
false

你可以像这样检查它的类型：

julia> typeof(fit)
StatsModels.DataFrameRegressionModel{GLM.LinearModel{GLM.LmResp{Array{Float64,1}},GLM.DensePredChol{Float64,Base.LinAlg.Cholesky{Float64,Array{Float64,2}}}},Array{Float64,2}}

附带说明：fit 不是一个非常幸运的名称，因为它与 StatsBase 导出的函数名称冲突。