如何从Julia GLM对象中提取模型矩阵

Question

我正在使用GLM包在 Julia 中进行线性回归。我有兴趣从 LM 对象中获取模型矩阵，这样我就可以计算杠杆值（帽子矩阵的对角线），但是找不到任何函数来执行此操作。 R中的等价物是model.matrix。

• 有什么建议吗？

我想我可以通过矩阵乘法手动进行回归，但并不想重新发明轮子。

Answer 1

通过在LM对象（我称之为names(OLS)）上调用OLS来解决这个问题。然后可以通过OLS.mm提取模型矩阵。

Answer 2

如果您使用的数据是DataFrame，则可以使用以下内容：

using DataFrames, GLM
dat = dataset("car","Vocab")
x = ModelMatrix(ModelFrame(Vocabulary~Year+Sex+Education,dat)).m