R 的 e1071 包中的 Predict.svm 函数

Question

我正在使用 R 的 e1071 包中的 svm 函数。使用 svm 函数我可以获得 74% 的数据准确率，但是当我尝试将 predict 函数用于测试数据集时，它会返回所有两个类别之一中的数据，在另一个类别中显示 0。这就是代码的样子

test <- read.table(file.choose(), header=T)
pred <- predict(modelb, test)
summary(pred)
0   1 
101   0 

Answer 1

其中一种可能是您的训练数据集不平衡。

假设您有 100 个样本属于 0 类，1 个样本属于 1 类。在某些情况下，最好的解决方案是将每个新样本都放在 0 类中。

一些解决方案包括：

1. 处理平衡数据集
2. 为类分配不同的权重。我使用的是 kernlab 包，但我认为代码并没有那么不同，并且在计算模型时应该有一个选项 class.weight。