如何提取Julia
中一般线性模型中指定的数据分布?例如,下面是一个玩具样例泊松GLM。我想从模型中提取一个表示"泊松"的字符串。类似地,如果模型被指定为数据分布=Binomial()
,我希望字符串返回'Binomial'。
### Load packages required
using DataFrames
using GLM
### Simulate some data for a dummy GLM
data = DataFrame(X=[1,2,3,4,5,6,7,8,9,10], Y=[2,4,7,3,1,6,3,2,5,1])
### Fit Poisson GLM
m1 = fit(GeneralizedLinearModel,
@formula(Y ~ X),
data,
Poisson(),
LogLink())
最后,我想写一个函数,接受一个if
语句,其中if == Poisson
然后做X,else
做y
刚才Julia Discourse讨论过这个问题,你可以这样做:
julia> typeof(m1).parameters[1].parameters[1].parameters[2]
Poisson{Float64}