优化 TextRanker

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)