更改rpart图中分支名称的格式

# 安装并加载必要的包
install.packages("rpart")
install.packages("rpart.plot")
library(rpart)
library(rpart.plot)

# 构建一个简单的决策树模型
data(iris)
tree_model <- rpart(Species ~ ., data = iris)

# 自定义函数来更改分支名称
customize_labels <- function(tree_obj) {
  labels <- attr(tree_obj$frame, "names")
  new_labels <- gsub("Class:", "类别:", labels)
  return(new_labels)
}

# 使用自定义函数更改分支名称并绘制决策树
rpart.plot(tree_model, extra = 1, faclen = 0, 
           fallen.leaves = TRUE, 
           branch.lty = 3, 
           tweak = 1.2,
           under = TRUE,
           nn = TRUE,
           type = 3,
           clip.right.labs = FALSE,
           branch.col = "gray",
           split.col = "black",
           split.cex = 1,
           split.lty = 2,
           split.round = 0,
           leaf.round = 1,
           leaf.col = "white",
           box.palette = "GnBu",
           yshift = 0.1,
           xshift = 0.1,
           main = "自定义分支名称的决策树",
           sub = "",
           cex.main = 1.5,
           cex.sub = 1.2,
           cex.axis = 1.2,
           cex.lab = 1.2,
           cex = 1.2,
           labels = customize_labels(tree_model))