R中的kNN使用两个参数

时间:2017-11-16 01:44:13

标签: r knn

是否可以训练与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)。

1 个答案:

答案 0 :(得分:0)

解决方案是创建一个列,将两个值都作为一个并使用:

crabs$parameter <- paste(crabs$sp , crabs$sex, sep="|")