You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

tidymodels与bonsai中如何设置partykit::ctree的mincriterion和testtype参数?

问题解答

可以通过set_engine()传递引擎专属的控制参数来修改mincriterion和testtype,具体做法是在set_engine()中传入control参数,值为partykit::ctree_control()创建的配置对象,示例代码如下:

model_ctree <- decision_tree() %>% 
  set_mode("regression") %>% 
  set_engine("partykit", 
             control = partykit::ctree_control(mincriterion = 0.95, 
                                              testtype = "Bonferroni",
                                              maxdepth = 10)) %>% 
  fit(formula, data = sample)

说明

parsnip的decision_tree()接口仅暴露了少量通用参数,但它支持直接传递引擎(此处为partykit)的原生配置参数。对于partykit的条件树来说,那些原本在ctree_control()里设置的参数,都可以通过set_engine()的control参数传入,从而实现和直接调用partykit::ctree()完全一致的配置效果。

内容的提问来源于stack exchange,提问作者Buoyant Xu

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.20 12:42:25