使用稀疏矩阵

时间:2016-06-22 23:09:02

标签: r matrix machine-learning svm sparse-matrix

我已使用Matrix包将数据帧(10430 X 1415)转换为稀疏矩阵。我将数据采样到训练数据和测试数据中。我想使用稀疏矩阵构建一个SVM模型(' e1071')。任何人都可以帮助我得到结果。以下是我正在尝试的代码。

library('e1071')

svm.model<-svm(trainData[,"Target"] ~.,data= trainData,kernel='linear', scale=FALSE)
  

错误:&#34;不能强制类&#34;结构(&#34; dgCMatrix&#34;,包=&#34; Matrix&#34;)&#34;到data.frame&#34;

另外请建议如何在testData上使用预测功能。

1 个答案:

答案 0 :(得分:-1)

要创建稀疏矩阵,您可以执行以下操作:

library(Matrix)
library(SpareM)
sMatrix <- Matrix(data=as.matrix(trainData), sparse=TRUE)

然后你可以在你的svm模型上使用它

svm.model <- svm(trainData[,"Target"] ~., data=sMatrix, kernel="linear")

预测您应该确保您的列车数据格式正确,然后您可以使用:

predicted <- predict(svm.model, testData)