df14=rock数据集 怎么建模 代码 用r语言
建模步骤如下:
- 加载数据集
data(df14)
- 数据预处理
# 删除无用变量
df14 <- df14[, -c(1, 2, 3)]
# 将分类变量转换为因子
df14$Class <- as.factor(df14$Class)
- 划分训练集和测试集
set.seed(123)
train_index <- sample(nrow(df14), 0.7 * nrow(df14))
train_data <- df14[train_index, ]
test_data <- df14[-train_index, ]
- 建立模型
# 安装和加载rpart包
install.packages("rpart")
library(rpart)
# 建立决策树模型
model <- rpart(Class ~ ., data = train_data, method = "class")
- 模型评估
# 预测测试集
pred <- predict(model, test_data, type = "class")
# 计算混淆矩阵
library(caret)
confusionMatrix(pred, test_data$Class)
``
原文地址: https://www.cveoy.top/t/topic/gBir 著作权归作者所有。请勿转载和采集!