R中的Dictionary()函数问题

时间:2017-11-06 16:24:37

标签: r dictionary

根据Lantz的书“机器学习用R”,我一直在关注贝叶斯分类器的例子。该案例是垃圾邮件分类器,它使用以下链接的数据:

http://www.dt.fee.unicamp.br/~tiago/smsspamcollection/

在代码中我遇到了这个问题:

sms_train<-DocumentTermMatrix(sms_corpus_train,list(dictionary=sms_dict))
sms_test<-DocumentTermMatrix(sms_corpus_test,list(dictionary=sms_dict))

因为它说我应该使用以下指令:

sms_dict <- Dictionary(findFreqTerms(sms_dtm_train, 5))

问题是从新版本的tm中不推荐使用Dictionary()函数。我应该怎样做才能完成这些书的内容:

  

字典是一种数据结构,允许我们指定哪些单词   应出现在文档术语矩阵中。限制我们的培训和   测试矩阵只到前面词典中的单词,使用   以下命令

我做了以下事情:

sms_dict<-findFreqTerms(sms_dtm_train,5)
sms_train<-DocumentTermMatrix(sms_corpus_train,list(dictionary=sms_dict))
sms_test<-DocumentTermMatrix(sms_corpus_test,list(dictionary=sms_dict))

但我相信我并没有限制测试矩阵。尽管代码工作正常,但它并没有给我正确的结果。在这种情况下我可以修改什么?

用于跟踪目的的完整代码如下:

sms_raw<-read.csv("sms_spam.csv",stringsAsFactors=FALSE)
install.packages("tm")
library(tm)
sms_corpus<-Corpus(VectorSource(sms_raw$text))
corpus_clean<-tm_map(sms_corpus,content_transformer(tolower))
corpus_clean<-tm_map(corpus_clean,removeNumbers)
corpus_clean<-tm_map(corpus_clean,removeWords,stopwords())
corpus_clean<-tm_map(corpus_clean,stripWhitespace)
sms_dtm<-DocumentTermMatrix(corpus_clean)
sms_raw_train<-sms_raw[1:4169,]
sms_raw_test<-sms_raw[4170:5559,]
sms_dtm_train<-sms_dtm[1:4169,]
sms_dtm_test<-sms_dtm[4170:5559,]
sms_corpus_train<-corpus_clean[1:4169]
sms_corpus_test<-corpus_clean[4170:5559]
sms_dict<-findFreqTerms(sms_dtm_train,5)
sms_train<-DocumentTermMatrix(sms_corpus_train,list(dictionary=sms_dict))
sms_test<-DocumentTermMatrix(sms_corpus_test,list(dictionary=sms_dict))
convert_counts<-function(x){
x<-ifelse(x>0,1,0)
x<-factor(x,levels=c(0,1),labels=c("No","Yes"))
return(x)
}
sms_train<-apply(sms_train,MARGIN=2,convert_counts)
sms_test<-apply(sms_test,MARGIN=2,convert_counts)
library(e1071)
sms_classifier<-naiveBayes(sms_train,sms_raw_train$type)
sms_test_pred<-predict(sms_classifier,sms_test)
install.packages("gmodels")
library(gmodels)
CrossTable(sms_test_pred,sms_raw_test$type,prop.chisq=FALSE,prop.t=FALSE,dnn=c('predicted','actual'))

由于

1 个答案:

答案 0 :(得分:0)

我遇到了同样的问题,并通过执行以下操作解决了该问题:

CrossTable(sms_test_pred[["class"]], sms_raw_test$Type, prop.chisq = FALSE, prop.t = FALSE, dnn = c('predicted','actual'))