Carlos-Francisco Méndez-Cruz

Feature extraction and vectorizer three sentences

......@@ -76,7 +76,7 @@ if __name__ == "__main__":
vectorizer = CountVectorizer(ngram_range=(1, 1))
else:
# Binary vectorizer
vectorizer = TfidfVectorizer(ngram_range=(1, 1), smooth_idf=False, norm=None)
vectorizer = TfidfVectorizer(ngram_range=(1, 1), norm=None)
matrix = csr_matrix(vectorizer.fit_transform(documents), dtype='double')
print(' matrix.shape: ', matrix.shape)
......