mirror of
https://github.com/fumiama/jieba.git
synced 2026-06-11 12:20:24 +08:00
small code refactor
This commit is contained in:
@@ -15,6 +15,8 @@ import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
var T *Trie
|
||||
|
||||
type Trie struct {
|
||||
Nodes mapset.Set
|
||||
MinFreq float64
|
||||
@@ -181,7 +183,7 @@ func addWord(word string, freq float64, tag string) {
|
||||
if len(tag) > 0 {
|
||||
UserWordTagTab[word] = strings.TrimSpace(tag)
|
||||
}
|
||||
trie.addWord(word, freq)
|
||||
T.addWord(word, freq)
|
||||
}
|
||||
|
||||
func LoadUserDict(filePath string) error {
|
||||
@@ -210,3 +212,8 @@ func LoadUserDict(filePath string) error {
|
||||
|
||||
return scanner.Err()
|
||||
}
|
||||
|
||||
func SetDictionary(dict_path string) (err error) {
|
||||
T, err = newTrie(dict_path)
|
||||
return
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user