这让我发疯了,我整天都在浏览类似的帖子,但似乎无法解决我的问题。我有一个朴素贝叶斯模型训练和存储为model
. 我正在尝试使用newdata
数据框进行预测,但我不断收到错误消息 Error: $ operator is invalid for atomic vectors
。这是我正在运行的:另一个数据框的第一行在stats::predict(model, newdata = newdata)
哪里newdata
:new data <- pbp[1, c("balls", "strikes", "outs_when_up", "stand", "pitcher", "p_throws", "inning")]
class(newdata)
给[1] "tbl_df" "tbl" "data.frame"
.
问题在于使用的数据。它应该与levels
训练中使用的匹配。例如,如果我们使用从 trainingData 到 的行之一predict
,它确实有效
predict(model, head(model$trainingData, 1))
#[1] Curveball
#Levels: Changeup Curveball Fastball Sinker Slider
通过检查str
两个数据集的 ,factor
训练中的一些列是character
类
str(model$trainingData)
'data.frame': 1277525 obs. of 7 variables:
$ pitcher : Factor w/ 1390 levels "112526","115629",..: 277 277 277 277 277 277 277 277 277 277 ...
$ stand : Factor w/ 2 levels "L","R": 1 1 2 2 2 2 2 1 1 1 ...
$ p_throws : Factor w/ 2 levels "L","R": 2 2 2 2 2 2 2 2 2 2 ...
$ balls : num 0 1 0 1 2 2 2 0 0 0 ...
$ strikes : num 0 0 0 0 0 1 2 0 1 2 ...
$ outs_when_up: num 1 1 1 1 1 1 1 2 2 2 ...
$ .outcome : Factor w/ 5 levels "Changeup","Curveball",..: 3 4 1 4 1 5 5 1 1 5 ...
str(newdata)
tibble [1 × 6] (S3: tbl_df/tbl/data.frame)
$ balls : int 3
$ strikes : int 2
$ outs_when_up: int 1
$ stand : chr "R"
$ pitcher : int 605200
$ p_throws : chr "R"
一个选择是levels
为factor
班级做同样的事情
nm1 <- intersect(names(model$trainingData), names(newdata))
nm2 <- names(which(sapply(model$trainingData[nm1], is.factor)))
newdata[nm2] <- Map(function(x, y) factor(x, levels = levels(y)), newdata[nm2], model$trainingData[nm2])
现在做predict
离子
predict(model, newdata)
#[1] Sinker
#Levels: Changeup Curveball Fastball Sinker Slider
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句