From 646f61b24a8fd894e329c3be43bc0d4be3145965 Mon Sep 17 00:00:00 2001 From: abtion Date: Mon, 22 Jun 2020 18:07:20 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8F=A5=E5=AD=90=E6=95=B0?= =?UTF-8?q?=E9=87=8F=E5=B0=8F=E4=BA=8Etopsen=E6=97=B6=E6=8A=A5=E9=94=99?= =?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- jiagu/textrank.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]