使用partykit决策树时遇「无效区间数量」错误的排查求助

尝试在partykit包的ctree中实现强制首次按split.variable拆分的逻辑时，触发如下错误：

Error in interval.numeric(x, breaks = c(xmin - tol, ux, xmax)) :
invalid number of intervals

数据特性：

• target为待预测分类变量
• split.variable为分类变量：当它取1时，target必为1；取0时，target可为0或1

已尝试将变量转为因子/数值类型、子采样数据，但问题仍未解决，以下是可复现错误的最小示例（MRE）：

library(partykit)

tdf = structure(list(target = c(0, 0, 0, 1, 0, 0, 1, 1, 1, 1), split.variable = c(0, 
0, 0, 0, 1, 0, 0, 0, 0, 0), var1 = c(2.021, 1.882, 1.633, 3.917, 
2.134, 1.496, 1.048, 1.552, 1.65, 3.112), var2 = c(97.979, 98.118, 
98.367, 96.083, 97.866, 98.504, 98.952, 98.448, 98.35, 96.888
), var3 = c(1, 1, 1, 0.98, 1, 1, 1, 1, 1, 1), var4 = c(1, 1, 
1, 0.98, 1, 1, 1, 1, 1, 1), var5 = c(18.028, 25.207, 20.788, 
28.548, 18.854, 19.984, 27.352, 24.622, 25.037, 24.067), var6 = c(0.213, 
0.244, 0.289, 0.26, 0.887, 0.575, 0.097, 0.054, 0.104, 0.096), 
    var7 = c(63.22, 59.845, 62.45, 63.48, 52.143, 51.256, 56.296, 
    57.494, 59.543, 68.434), var8 = c(0.748, 0.795, 0.807, 0.793, 
    0.901, 0.909, 0.611, 0.61, 0.618, 0.589)), row.names = c(6L, 
7L, 8L, 9L, 11L, 12L, 15L, 16L, 17L, 18L), class = "data.frame")

tr1 <- ctree(target ~ split.variable,     data = tdf, maxdepth = 1)
tr2 <- ctree(target ~ split.variable + ., data = tdf, subset = predict(tr1, type = "node") == 2)

错误原因 #

在你的MRE中：

1. tr1生成的树里，split.variable=1对应的是节点1（该节点所有target都是1，是纯节点），剩余样本（split.variable=0）对应节点2
2. 当你用predict(tr1, type="node") == 2筛选子集后，这个子集中的split.variable只有0这一个取值
3. 此时在tr2的公式中仍保留split.variable，ctree会尝试对这个无变异的变量做拆分，自然触发"invalid number of intervals"错误——因为无法对单一取值的变量生成有效拆分区间

修正步骤 #

1. 先确认tr1的节点分配（可选，但能帮你明确子集范围）：

print(predict(tr1, type = "node"))

输出会显示：split.variable=1的样本对应节点1，其余为节点2。

2. 调整tr2的公式，移除split.variable（因为子集里它无变异，无法拆分）：

# 直接用"."代表除target外的所有变量（自动排除split.variable，因为子集里它无变异会被忽略）
tr2 <- ctree(target ~ ., data = tdf, subset = predict(tr1, type = "node") == 2)

完整可运行代码 #

library(partykit)

tdf = structure(list(target = c(0, 0, 0, 1, 0, 0, 1, 1, 1, 1), split.variable = c(0, 
0, 0, 0, 1, 0, 0, 0, 0, 0), var1 = c(2.021, 1.882, 1.633, 3.917, 
2.134, 1.496, 1.048, 1.552, 1.65, 3.112), var2 = c(97.979, 98.118, 
98.367, 96.083, 97.866, 98.504, 98.952, 98.448, 98.35, 96.888
), var3 = c(1, 1, 1, 0.98, 1, 1, 1, 1, 1, 1), var4 = c(1, 1, 
1, 0.98, 1, 1, 1, 1, 1, 1), var5 = c(18.028, 25.207, 20.788, 
28.548, 18.854, 19.984, 27.352, 24.622, 25.037, 24.067), var6 = c(0.213, 
0.244, 0.289, 0.26, 0.887, 0.575, 0.097, 0.054, 0.104, 0.096), 
    var7 = c(63.22, 59.845, 62.45, 63.48, 52.143, 51.256, 56.296, 
    57.494, 59.543, 68.434), var8 = c(0.748, 0.795, 0.807, 0.793, 
    0.901, 0.909, 0.611, 0.61, 0.618, 0.589)), row.names = c(6L, 
7L, 8L, 9L, 11L, 12L, 15L, 16L, 17L, 18L), class = "data.frame")

# 生成首次拆分的树
tr1 <- ctree(target ~ split.variable, data = tdf, maxdepth = 1)
# 查看节点分配
print(predict(tr1, type = "node"))

# 对节点2的子集构建后续树，移除无变异的split.variable
tr2 <- ctree(target ~ ., data = tdf, subset = predict(tr1, type = "node") == 2)
# 输出结果
print(tr2)

内容的提问来源于stack exchange，提问作者Nico