From ed0bc44921782b89a069d93772a29e468167a634 Mon Sep 17 00:00:00 2001
From: Wang Bin <wangbin.zibo@gmail.com>
Date: Thu, 26 Feb 2015 17:45:59 +0800
Subject: [PATCH] removed third party dependency

---
 analyse/textrank.go | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/analyse/textrank.go b/analyse/textrank.go
index ed39f02..c48468a 100644
--- a/analyse/textrank.go
+++ b/analyse/textrank.go
@@ -2,7 +2,6 @@ package analyse
 
 import (
 	"fmt"
-	mapset "github.com/deckarep/golang-set"
 	"github.com/wangbin/jiebago/posseg"
 	"math"
 	"sort"
@@ -117,21 +116,21 @@ func (u *undirectWeightedGraph) rank() TfIdfs {
 }
 
 func TextRankWithPOS(sentence string, topK int, allowPOS []string) TfIdfs {
-	posFilt := mapset.NewSet()
+	posFilt := make(map[string]int)
 	for _, pos := range allowPOS {
-		posFilt.Add(pos)
+		posFilt[pos] = 1
 	}
 	g := newUndirectWeightedGraph()
 	cm := make(map[[2]string]float64)
 	span := 5
 	wordTags := posseg.Cut(sentence, true)
 	for i, _ := range wordTags {
-		if posFilt.Contains(wordTags[i].Tag) {
+		if _, ok := posFilt[wordTags[i].Tag]; ok {
 			for j := i + 1; j < i+span; j++ {
 				if j > len(wordTags) {
 					break
 				}
-				if !posFilt.Contains(wordTags[j].Tag) {
+				if _, ok := posFilt[wordTags[j].Tag]; !ok {
 					continue
 				}
 				if _, ok := cm[[2]string{wordTags[i].Word, wordTags[j].Word}]; !ok {