我正在尝试使用knn算法进行分类。我的问题是如何调整算法使用的邻居数量?
例如,我想使用3、9和12?
如何在命令中调整此参数?
species_knn = train(species ~., method= "knn", data = species, trControl=trainControl(method = 'cv', number = 3))发布于 2018-01-07 22:34:35
下面是一个使用虹膜数据的网格搜索示例:
library(caret)构建要调整的超参数网格:
grid = expand.grid(k = c(3, 9, 12)) #in this case data.frame(k = c(3, 9, 12)) will do在tuneGrid参数中提供网格:
species_knn = train(Species ~., method= "knn",
data = iris,
trControl = trainControl(method = 'cv',
number = 3,
search = "grid"),
tuneGrid = grid)
species_knn$results
#output
k Accuracy Kappa AccuracySD KappaSD
1 3 0.9666667 0.9499560 0.02309401 0.0346808964
2 9 0.9600000 0.9399519 0.00000000 0.0000416525
3 12 0.9533333 0.9299479 0.01154701 0.0173066504Here是所有可用型号和超参数的列表。
https://stackoverflow.com/questions/48138031
复制相似问题