r - 机器学习: predicting a new value in R

Question

只需遵循 ML 教程即可 ( https://machinelearningmastery.com/machine-learning-in-r-step-by-step/ )。我正在尝试预测单个向量的单个输出。

我正在使用 IRIS 数据集，并尝试使用 LDA 算法进行预测。我正在尝试使用我的 nval 向量 (c(4.3,3.1,1.5,0.1) 来预测单个结果，它是什么物种？

fit.lda <- train(Species~., data=dataset, method="lda", metric=metric, 
trControl=control)
predictions <- predict(fit.lda, validation)

这会产生预期的常规输出。

predict(fit.lda, nval)

给我:

警告消息:“newdata”有 4 行，但找到的变量有 120 行

帮助:)

Answer 1

library(caret)
control <- trainControl(method="cv", number=10)
metric <- "Accuracy"

validation_index <- createDataPartition(iris$Species, p=0.80, list=FALSE)
validation <- iris[-validation_index,]
dataset <- iris[validation_index,]  

fit.lda <- train(Species~., data=dataset, method="lda", metric=metric, 
                 trControl=control)

您需要将 nval 定义为数据框，其中列的名称如 iris

nval <- data.frame(4.3, 3.1, 1.5, 0.1)
names(nval) <- names(iris)[1:4]
predict(fit.lda, nval)

# [1] setosa