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
相关产品推荐
相关产品推荐

