能否用lmtree拟合多响应变量线性模型？正确语法是什么？

架构老林

2026-6-13

如何用lmtree拟合含多响应变量的模型？

lmtree本身不支持直接通过cbind(Y1,Y2)的方式指定多响应变量，但可以通过以下两种方式实现需求：

方法1：使用`mlmtree`（推荐）

partykit包中的mlmtree函数专门为多响应线性模型树设计，完美适配你的场景，不会出现下标错误：

library(partykit)
mlmtree(cbind(Y1, Y2) ~ Z1 | X1 + X2 + X3, data = my_df)

方法2：自定义拟合函数适配`lmtree`

如果一定要使用lmtree，可以将多响应变量整合成矩阵，并自定义拟合函数：

# 自定义支持多响应的lm拟合函数
multi_lm <- function(formula, data, ...) {
  lm(formula, data = data, ...)
}

# 转换数据并调用lmtree
my_df$Y_matrix <- cbind(my_df$Y1, my_df$Y2)
lmtree(Y_matrix ~ Z1 | X1 + X2 + X3, data = my_df, fit = multi_lm)

注意：这种方法在后续的树解析、预测等环节可能需要额外处理，不如mlmtree便捷。