更改 x-axis 限制 partykit::lmtree (mob) plots

Change x-axis limits for partykit::lmtree (mob) plots

如标​​题所示:我想更改 partykit::lmtree (mob) 个图的 x-axis 限制。

玩具示例

Y~X+X^2 回归的形状取决于 A:

A <- rep(1:5, each=4)
X <- 1:20
Y <- X*A+(A<3)*X^2+rnorm(20, mean=10)

现在用多项式项和绘图构建 lmtree

library(partykit)
mytree <- lmtree(Y~poly(X,2,raw=T) | A, minsize=3)
plot(mytree)

要得到这个:

X-axis 的限制显然基于 poly(X,2,raw=T) 的范围,而不是 X 的范围。我该如何更改它们?

我试过的

目前,xscaleyscale 参数(与基本图形参数 xlimylim 等效的网格图形)在 node_bivplot() 内部设置并且不能通过参数修改。

在你的情况下,我建议手动设置多项式的两个回归量 (X + I(X^2)),然后只为这两个回归量中的第一个创建一个终端面板 (which = 1):

mytree <- lmtree(Y ~ X + I(X^2) | A, minsize = 3)
plot(mytree, tp_args = list(which = 1))