michaelkri commited on
Commit
ac34fec
·
1 Parent(s): 037df93

Changed default parameter

Browse files
Files changed (1) hide show
  1. app/summarizer.py +3 -3
app/summarizer.py CHANGED
@@ -74,7 +74,7 @@ class Summarizer:
74
  return ranked_sentences
75
 
76
 
77
- def summarize_clusters(self, clusters, top_cluster_count=20):
78
  # sort clusters by their length (descending) to find the most important topics
79
  clusters = sorted(clusters, key=len, reverse=True)
80
 
@@ -86,7 +86,7 @@ class Summarizer:
86
  for i, cluster in enumerate(clusters):
87
  logging.debug(f'Extracting from cluster {i + 1}...')
88
  top_sentences = Summarizer.rank_cluster_sentences(cluster)
89
- content = '\n'.join(top_sentences[:2])
90
  key_sentences.append(content)
91
 
92
  combined = ' '.join(key_sentences)
@@ -101,7 +101,7 @@ class Summarizer:
101
  return summary
102
 
103
 
104
- def multisource_summary(self, articles, min_cluster_size=3):
105
  '''
106
  Create a single summary from multiple articles
107
  '''
 
74
  return ranked_sentences
75
 
76
 
77
+ def summarize_clusters(self, clusters, top_cluster_count=10, top_k_sentences=10):
78
  # sort clusters by their length (descending) to find the most important topics
79
  clusters = sorted(clusters, key=len, reverse=True)
80
 
 
86
  for i, cluster in enumerate(clusters):
87
  logging.debug(f'Extracting from cluster {i + 1}...')
88
  top_sentences = Summarizer.rank_cluster_sentences(cluster)
89
+ content = '\n'.join(top_sentences[:top_k_sentences])
90
  key_sentences.append(content)
91
 
92
  combined = ' '.join(key_sentences)
 
101
  return summary
102
 
103
 
104
+ def multisource_summary(self, articles, min_cluster_size=2):
105
  '''
106
  Create a single summary from multiple articles
107
  '''