
Untitled
By: a guest on
Jun 27th, 2012 | syntax:
None | size: 0.55 KB | hits: 9 | expires: Never
def build_matrix(corpus, tfidf, dense_matrix):
print "Building term-document matrix for corpus"
if tfidf and dense_matrix:
try:
sim = similarities.MatrixSimilarity(tfidf[corpus])
except Exception, e:
import time
time.sleep(20)
elif tfidf and not dense_matrix:
sim = similarities.SparseMatrixSimilarity(tfidf[corpus])
elif dense_matrix:
sim = similarities.MatrixSimilarity(corpus)
else:
sim = similarities.SparseMatrixSimilarity(corpus)
return sim