diff --git a/jiagu/textrank.py b/jiagu/textrank.py index 17b7fa8..ec8fe1e 100644 --- a/jiagu/textrank.py +++ b/jiagu/textrank.py @@ -115,7 +115,7 @@ class Summarize(object): scores = utils.weight_map_rank(graph, self.__max_iter, self.__tol) sent_selected = nlargest(n, zip(scores, count())) sent_index = [] - for i in range(n): + for i in range(min(n, len(sent_selected))): sent_index.append(sent_selected[i][1]) return [sentences[i] for i in sent_index]