gpt4 book ai didi

R 指标 RMSE 不适用于分类模型

转载 作者:行者123 更新时间:2023-12-01 23:31:49 32 4
gpt4 key购买 nike

我正在尝试使用 R 和 xgboost 来研究我的模型。训练模型总体上效果很好,但对于插入符来说,度量存在一些问题。

我尝试为类列设置一个因子,但仍然没有结果。

我的数据

ID  var1var2TARGET
1 5 0 1
2 4 3 1
3 4 2 0
4 3 1 0
5 2 4 1
6 1 2 1
7 5 3 1
8 4 1 0
9 4 1 0
10 2 4 1
11 5 5 1

为此我愿意

train <- read.csv()
train.y <- train$TARGET
train$TARGET <- NULL
train$ID <- NULL
train.y <- lapply(train.y, factor)

然后我准备模型参数

xgb_grid_1 = expand.grid(
nrounds = 1000,
eta = c(0.01, 0.001, 0.0001),
max_depth = c(2, 4, 6, 8, 10),
gamma = 1
)

# pack the training control parameters
xgb_trcontrol_1 = trainControl(
method = "cv",
number = 5,
verboseIter = TRUE,
returnData = FALSE,
returnResamp = "all", # save losses across all models
classProbs = TRUE, # set to TRUE for AUC to be computed
summaryFunction = twoClassSummary,
allowParallel = TRUE
)

之后,我调用了训练函数

xgb_train_1 = train(
x = train,
y = train.y,
trControl = xgb_trcontrol_1,
tuneGrid = xgb_grid_1,
method = "xgbTree"
)

它给了我

Error in train.default(x = train, y = train.y, trControl = xgb_trcontrol_1,  : 
Metric RMSE not applicable for classification models

为什么会这样?

最佳答案

你应该尝试改变train.y <- lapply(train.y, factor)train.y <- factor(train.y, labels = c("yes", "no")) .

caret通常会提示标签为 0 或 1,因此请尝试更改它。

关于R 指标 RMSE 不适用于分类模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36665045/

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com