Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
@@ -42,12 +42,9 @@ def smart_summarize(text, n_clusters=1):
|
|
42 |
idx = np.where(kmeans.labels_ == i)[0]
|
43 |
if not len(idx):
|
44 |
continue
|
45 |
-
|
46 |
-
avg_vector = tfidf_matrix[idx].mean(axis=0).A # Convert matrix to ndarray
|
47 |
-
# Compute cosine similarity between avg_vector and tfidf vectors in cluster
|
48 |
sim = cosine_similarity(avg_vector, tfidf_matrix[idx].toarray())
|
49 |
most_representative = sentences[idx[np.argmax(sim)]]
|
50 |
summary_sentences.append(most_representative)
|
51 |
|
52 |
-
# Preserve original sentence order
|
53 |
return " ".join(sorted(summary_sentences, key=sentences.index))
|
|
|
42 |
idx = np.where(kmeans.labels_ == i)[0]
|
43 |
if not len(idx):
|
44 |
continue
|
45 |
+
avg_vector = np.asarray(tfidf_matrix[idx].mean(axis=0))
|
|
|
|
|
46 |
sim = cosine_similarity(avg_vector, tfidf_matrix[idx].toarray())
|
47 |
most_representative = sentences[idx[np.argmax(sim)]]
|
48 |
summary_sentences.append(most_representative)
|
49 |
|
|
|
50 |
return " ".join(sorted(summary_sentences, key=sentences.index))
|