From a8d1e81f73ac172c198e0c0b7ac772d226b8266d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Wed, 30 Nov 2022 12:56:04 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=20TextRanker?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 analyse/example_test.go  | 13 +++++++------
 analyse/textrank.go      | 12 +++++-------
 analyse/textrank_test.go |  6 ++++--
 3 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/analyse/example_test.go b/analyse/example_test.go
index af59fb4..b82f2e6 100755
--- a/analyse/example_test.go
+++ b/analyse/example_test.go
@@ -1,13 +1,11 @@
-package analyse_test
+package analyse
 
 import (
 	"fmt"
-
-	"github.com/fumiama/jieba/analyse"
 )
 
 func Example_extractTags() {
-	var t analyse.TagExtracter
+	var t TagExtracter
 	t.LoadDictionary("../dict.txt")
 	t.LoadIdf("idf.txt")
 
@@ -22,8 +20,11 @@ func Example_extractTags() {
 }
 
 func Example_textRank() {
-	var t analyse.TextRanker
-	t.LoadDictionary("../dict.txt")
+	t, err := LoadDictionary("../dict.txt")
+	if err != nil {
+		panic(err)
+	}
+
 	sentence := "此外，公司拟对全资子公司吉林欧亚置业有限公司增资4.3亿元，增资后，吉林欧亚置业注册资本由7000万元增加到5亿元。吉林欧亚置业主要经营范围为房地产开发及百货零售等业务。目前在建吉林欧亚城市商业综合体项目。2013年，实现营业收入0万元，实现净利润-139.13万元。"
 
 	result := t.TextRank(sentence, 10)
diff --git a/analyse/textrank.go b/analyse/textrank.go
index a895558..e84b838 100755
--- a/analyse/textrank.go
+++ b/analyse/textrank.go
@@ -132,7 +132,7 @@ func (t *TextRanker) TextRankWithPOS(sentence string, topK int, allowPOS []strin
 	}
 	span := 5
 	var pairs []posseg.Segment
-	for pair := range t.seg.Cut(sentence, true) {
+	for pair := range (*posseg.Segmenter)(t).Cut(sentence, true) {
 		pairs = append(pairs, pair)
 	}
 	for i := range pairs {
@@ -169,12 +169,10 @@ func (t *TextRanker) TextRank(sentence string, topK int) Segments {
 }
 
 // TextRanker is used to extract tags from sentence.
-type TextRanker struct {
-	seg *posseg.Segmenter
-}
+type TextRanker posseg.Segmenter
 
 // LoadDictionary reads a given file and create a new dictionary file for Textranker.
-func (t *TextRanker) LoadDictionary(fileName string) error {
-	t.seg = new(posseg.Segmenter)
-	return t.seg.LoadDictionary(fileName)
+func LoadDictionary(fileName string) (TextRanker, error) {
+	seg := posseg.Segmenter{}
+	return TextRanker(seg), seg.LoadDictionary(fileName)
 }
diff --git a/analyse/textrank_test.go b/analyse/textrank_test.go
index 339e3cf..a37fdc2 100755
--- a/analyse/textrank_test.go
+++ b/analyse/textrank_test.go
@@ -23,8 +23,10 @@ var (
 )
 
 func TestTextRank(t *testing.T) {
-	var tr TextRanker
-	tr.LoadDictionary("../dict.txt")
+	tr, err := LoadDictionary("../dict.txt")
+	if err != nil {
+		t.Fatal(err)
+	}
 	results := tr.TextRank(sentence, 10)
 	for index, tw := range results {
 		if tw.text != tagRanks[index].text || math.Abs(tw.weight-tagRanks[index].weight) > 1e-6 {