R的e1071軟件包中的Predict.svm函數

Question

我在R的e1071軟件包中使用了svm函數。使用svm函數，我的數據可以達到74％的精度，但是當我嘗試對測試數據集使用預測函數時，它會將所有數據返回一個兩個類別中的一個，在另一個類別中顯示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選項。