是否可以训练与trainData $ sp和trainData $ sex相关的数据?
library(dplyr)
library(caret)
library(e1071)
data(crabs, package = "MASS")
crabs = mutate_if(crabs, is.character, as.factor)
set.seed(1234)
index <- createDataPartition(crabs$sp, p=0.70, list=FALSE)
trainData= crabs[index,]
testData= crabs[-index,]
model_knn1 = knn3(trainData[,4:8], testData$sp, cl=trainData$sp, k=1)
在上面的代码中,我只训练与物种有关的数据(trainData $ sp)。
答案 0 :(得分:0)
解决方案是创建一个列,将两个值都作为一个并使用:
crabs$parameter <- paste(crabs$sp , crabs$sex, sep="|")