首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >k-NN示例中的错误精度度量

k-NN示例中的错误精度度量
EN

Stack Overflow用户
提问于 2017-03-19 10:28:44
回答 1查看 237关注 0票数 1

我是新接触k-NN的,所以我试图通过Datacamp课程: ML入门来学习UCI ML repo中的泰坦尼克型数据集。

我已经讲到了基于精度选择最佳k参数的部分(accs)。然而,当我在Rstudio上复制这个练习时,我得到了k=1;答案应该是k=73。我哪里错了?

代码语言:javascript
复制
#load dataset
titanic_train<-read.csv("https://kaggle2.blob.core.windows.net/competitions-data/kaggle/3136/train.csv?sv=2015-12-11&sr=b&sig=coHTg7HQb86RdaxfD2f9SiN492A4XLIDDRKVxLvw8Ys%3D&se=2017-03-19T10%3A26%3A52Z&sp=r")

# Omit NAs
titanic_train<-na.omit(titanic_train[,c("Survived", "Pclass", "Age", "Sex")])
library(class)
library(dplyr)

set.seed(1)
n <- nrow(titanic_train)
shuffled <- titanic_train[sample(n), ]

# Split the data : train and test
train_indices <- 1:round(0.7 * n)
train <- shuffled[train_indices, ]
test_indices <- (round(0.7 * n) + 1):n
test <- shuffled[test_indices, ]

# Store the Survived column of train and test in train_labels and test_labels
train_labels<-train$Survived
test_labels<-test$Survived

# Copy train and test to knn_train and knn_test
train$Sex = as.numeric(factor(train$Sex))
train <- train %>% mutate(Sex=ifelse(Sex==2, 1,0))
train$Survived=factor(train$Survived)
summary(train$Sex)
knn_train<-train
knn_train$Age<-round(knn_train$Age,7)

test$Sex = as.numeric(factor(test$Sex))
test <- test %>% mutate(Sex=ifelse(Sex==2, 1,0))
test$Survived=factor(test$Survived)
summary(test$Sex)
knn_test<-test

# Drop Survived column for knn_train and knn_test
knn_train$Survived<-NULL
knn_test$Survived<-NULL

# Normalize Pclass
min_class <- min(knn_train$Pclass)
max_class <- max(knn_train$Pclass)
knn_train$Pclass <- (knn_train$Pclass - min_class) / (max_class - min_class)
knn_test$Pclass <- (knn_test$Pclass - min_class) / (max_class - min_class)

# Normalize Age
min_age <- min(knn_train$Age)
max_age <- max(knn_train$Age)
knn_train$Age <- (knn_train$Age-min_age)/(max_age-min_age)
knn_test$Age <- (knn_test$Age-min_age)/(max_age-min_age)
summary(train)

# Set random seed.
set.seed(1)

# define range and accs
range <- 1:round(0.2 * nrow(knn_train))
accs <- rep(0, length(range))

for (k in range) {
  pred <- knn(knn_train, knn_test, train_labels, k = k)
  pred<-factor(pred, levels=c(1,0))
  test_labels<-factor(test_labels, levels=c(1,0))
  conf <- table(test_labels, pred)
  accs[k] <- sum(diag(conf)) / sum(conf)
}

# Plot the accuracies. 
plot(range, accs, xlab = "k")

# Calculate the best k
which.max(accs)
accs[which.max(accs)]
EN

回答 1

Stack Overflow用户

发布于 2017-03-19 12:18:22

我猜你想要最优的k计算,你把它定义为 k (最后一条评论:#计算最好的k)。

然而,调用它只会得到的最大

代码语言:javascript
复制
which.max(accs)
accs[which.max(accs)]

毫无疑问,k=1的最大准确率是0.8364486。

如果你想找到最优的k,你需要目视检查它(使用scree图)。找到最长的弯头,并设置k的值。(这是一个主观决定,因人而异。)

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/42882263

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档