R - partykit - 自定义模型树节点和拆分

Question

在partykit 包中，有一种方法可以通过指定预测器和拆分来构建自定义树。例如：

data("WeatherPlay", package = "partykit")

#create a split
sp_o <- partysplit(3L, breaks = 75) 

#create a node
n1 <- partynode(id = 1L, split = sp_o, kids = lapply(2L:3L, partynode))

#and make a "tree" out of it
t2 <- party(
  n1,
  data = WeatherPlay,
  fitted = data.frame(
    "(fitted)" = fitted_node(n1, data = WeatherPlay),
    "(response)" = WeatherPlay$play,
    check.names = FALSE
  ),
  terms = terms(play ~ ., data = WeatherPlay),
)

t2 <- as.constparty(t2)
t2
plot(t2)

这对于模型树（由 mob() 返回）是否可行？我可以逐个节点构建树节点，然后将指定的函数拟合到终端节点吗？

Answer 1

原则上，可以手动构建一个"modelparty" 对象（由mob() 返回）。但是，在您引用的示例中，没有像 as.constparty() 这样的简单强制函数。原因是为了构建模型树，以及打印，尤其是使用它们进行预测，需要更多关于模型的详细信息。

我建议先构建树形结构 ("partynode")，然后填充整个$info 槽（使用call、formula、Formula、terms、fit、@987654331 @、dots 和 nreg)。然后应该可以调用refit.modelparty() 来重新拟合所有终端节点中的模型。然后可以使用它来填充$node$info（带有coefficients、objfun、nobs，...）。

免责声明：所有这些都是完全未经测试的。但是，除了模仿"modelparty"，您当然也可以创建自己的方式将模型存储在"party" 对象中，并使用partykit 提供的更基本的构建块。