Hasitha16 commited on
Commit
356f40d
·
verified ·
1 Parent(s): e430cc4

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -4
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
- # Average vector from cluster, converted to ndarray
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))