1

我使用 LDA 为 2 个文本文档 A 和 B 建立主题模型。文档 A 与计算机科学高度相关，文档 B 与地球科学高度相关。然后我使用这个命令训练了一个 lda：

     text<- c(A,B) # introduced above
     r <- Corpus(VectorSource(text)) # create corpus object
     r <- tm_map(r, tolower) # convert all text to lower case
     r <- tm_map(r, removePunctuation) 
     r <- tm_map(r, removeNumbers)
     r <- tm_map(r, removeWords, stopwords("english"))
     r.dtm <- TermDocumentMatrix(r, control = list(minWordLength = 3))    
     my_lda <- LDA(r.dtm,2)

现在我想使用 my_lda 来预测一个新文档的上下文，比如 C，我想看看它是否与计算机科学或地球科学有关。我知道我是否使用此代码进行预测

     x<-C# a new document (a long string) introduced above for prediction
     rp <- Corpus(VectorSource(x)) # create corpus object
     rp <- tm_map(rp, tolower) # convert all text to lower case
     rp <- tm_map(rp, removePunctuation) 
     rp <- tm_map(rp, removeNumbers)
     rp <- tm_map(rp, removeWords, stopwords("english"))
     rp.dtm <- TermDocumentMatrix(rp, control = list(minWordLength = 3))    
     test.topics <- posterior(my_lda,rp.dtm)

它会给我一个标签 1 或 2，但我不知道 1 或 2 代表什么……我怎么知道它是指计算机科学相关还是地球科学相关？

4

1 回答 1

2

您可以从 LDA 主题模型中提取最可能的术语，并用您想要的任意数量替换这些黑盒数字名称。您的示例不可重现，但这里是说明如何执行此操作的示例：

> library(topicmodels)
> data(AssociatedPress)
> 
> train <- AssociatedPress[1:100]
> test <- AssociatedPress[101:150]
> 
> train.lda <- LDA(train,2)
> 
> #returns those black box names
> test.topics <- posterior(train.lda,test)$topics
> head(test.topics)
              1           2
[1,] 0.57245696 0.427543038
[2,] 0.56281568 0.437184320
[3,] 0.99486888 0.005131122
[4,] 0.45298547 0.547014530
[5,] 0.72006712 0.279932882
[6,] 0.03164725 0.968352746
> #extract top 5 terms for each topic and assign as variable names
> colnames(test.topics) <- apply(terms(train.lda,5),2,paste,collapse=",")
> head(test.topics)
     percent,year,i,new,last new,people,i,soviet,states
[1,]              0.57245696                0.427543038
[2,]              0.56281568                0.437184320
[3,]              0.99486888                0.005131122
[4,]              0.45298547                0.547014530
[5,]              0.72006712                0.279932882
[6,]              0.03164725                0.968352746
> #round to one topic if you'd prefer
> test.topics <- apply(test.topics,1,function(x) colnames(test.topics)[which.max(x)])
> head(test.topics)
[1] "percent,year,i,new,last"    "percent,year,i,new,last"    "percent,year,i,new,last"   
[4] "new,people,i,soviet,states" "percent,year,i,new,last"    "new,people,i,soviet,states"
于 2013-05-20T23:27:14.640 回答