Сравнить условия документов в матрице сроков документов в R
Мне нужно построить Матрицу сходства, сравнивая условия документов. Так, например, если Document1 и Document2 имеют 2 одинаковых термина, мне нужно написать 2 в моей матрице сходства в m[1, 2]. Моя матрица сходства выглядит следующим образом:
[,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9]
[1,] 0 NA NA NA NA NA NA NA NA
[2,] 0 0 NA NA NA NA NA NA NA
[3,] 0 0 0 NA NA NA NA NA NA
[4,] 0 0 0 0 NA NA NA NA NA
[5,] 0 0 0 0 0 NA NA NA NA
[6,] 0 0 0 0 0 0 NA NA NA
[7,] 0 0 0 0 0 0 0 NA NA
[8,] 0 0 0 0 0 0 0 0 NA
Документы и условия находятся в матрице условий документа. Теперь мне нужно заполнить матрицу сходства, сравнив все документы и их термины, где написано "NA" в матрице сходства. Для каждого совпадения терминов в паре документов мне нужно сосчитать +1 и ввести конечное значение в нужное место в матрице.
Моя проблема в том, что я не могу получить доступ к отдельным документам и их терминам в матрице терминов документа. Есть ли другой способ сделать это или я что-то упустил? Вот код:
install.packages("tm")
install.packages("openNLP")
install.packages("openNLPmodels.en")
Sys.setenv(NOAWT=TRUE)
library(tm)
library(openNLP)
library(openNLPmodels.en)
sample = c(
"count eagle alien",
"dis bound eagle",
"bound count eagle dis",
"count eagle dis alien",
"bound eagle",
"count dis alien",
"bound count alien",
"bound count",
"count eagle dis"
)
print(sample)
corpus <- Corpus(VectorSource(sample))
inspect(corpus)
corpus <- tm_map(corpus, removeNumbers)
corpus <- tm_map(corpus, removePunctuation)
corpus <- tm_map(corpus, tolower)
corpus <- tm_map(corpus, removeWords, stopwords("english"))
corpus <- tm_map(corpus, stemDocument,language="english")
corpus <- tm_map(corpus, stripWhitespace)
corpus <- tm_map(corpus, tmTagPOS)
inspect(corpus)
dtm <- DocumentTermMatrix(corpus)
inspect(dtm)
# need to create similarity matrix here
#dist(dtm, method = "manhattan", diag = FALSE, upper = TRUE)
rowCount <- nrow(dtm)
similMatrix = matrix(nrow = rowCount - 1, ncol = rowCount)
show(similMatrix)
similMatrix[ row(similMatrix) >= col(similMatrix) ] <- 0
for(i in 1:(rowCount - 1)){ # rows
for (j in i+1:rowCount){ # cols
# need to compare document i and j here and write
# the value into similarity matrix
}
}
show(similMatrix)
1 ответ
Я думаю, что вам не хватает еще одной строки в вашей матрице сходства. Потому что вы не покрыли свой последний документ. Моя выглядит так.
[,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9]
[1,] NA NA NA NA NA NA NA NA NA
[2,] 1 NA NA NA NA NA NA NA NA
[3,] 2 3 NA NA NA NA NA NA NA
[4,] 3 2 3 NA NA NA NA NA NA
[5,] 1 2 2 1 NA NA NA NA NA
[6,] 2 1 2 3 0 NA NA NA NA
[7,] 2 1 2 2 1 2 NA NA NA
[8,] 1 1 2 1 1 1 2 NA NA
[9,] 2 2 3 3 1 2 1 1 NA
Чтобы получить этот результат, я сделал следующие шаги.
mat=as.data.frame(as.matrix(dtm)) # you get the dataframe from DocumentTerm Matrix
rowCount <- nrow(dtm)
colCount <- ncol(dtm)
similMatrix = matrix(nrow = rowCount, ncol = rowCount)
similMatrix[ row(similMatrix) >= col(similMatrix) ] <- 0
for(i in 1:(rowCount)){ #set all columns NA you can change to zeros if you need later
similMatrix[i,i]=NA
} # then we will do the actual job
for(i in 1:rowCount ){ # rows
for (j in 1:rowCount ){ # cols
if(is.na(similMatrix[i,j])==F){
a=mat[i,]
b=mat[j,]
for(k in 1:colCount){ #n number of Cols in Document term matrix
if(a[k]==1 && a[k]==b[k]){
similMatrix[i,j]=similMatrix[i,j]+1
}
}
}
}
}