Skip to content

Latest commit

 

History

History
279 lines (206 loc) · 8.53 KB

File metadata and controls

279 lines (206 loc) · 8.53 KB

gse

Go による効率的な多言語 NLP およびテキスト分割ライブラリ。英語、中国語、日本語などをサポートします。 elasticsearchbleve もサポートしています。

Build Status CircleCI Status codecov Go Report Card GoDoc GitHub release Join the chat at https://gitter.im/go-ego/ego

Gse は jieba の Golang 実装であり、NLP サポートやより多くの機能を追加することを目指しています。

機能:

  • 一般、検索エンジン、完全、精密、HMM モードなど複数の分かち書きモードをサポート
  • ユーザー辞書と埋め込み辞書、品詞タグ付け、分割情報の解析、ストップワードと単語のトリムをサポート
  • 多言語サポート:英語、中国語、日本語など
  • 繁体字中国語をサポート
  • Viterbi アルゴリズムを使った HMM テキスト分割をサポート
  • TensorFlow による NLP サポート(開発中)
  • 固有表現認識(開発中)
  • elasticsearch と bleve をサポート
  • JSON RPC サービスの実行が可能

アルゴリズム:

  • 辞書は Double Array Trie(ダブル配列トライ)で実装
  • 分割器アルゴリズムは最短パス(単語頻度と動的プログラミングに基づく)、DAG および HMM アルゴリズムによる分かち書き

テキスト分割速度:

バインディング:

gse-bind、JavaScript やその他の言語をバインドし、より多くの言語をサポートします。

インストール / アップデート

Go モジュールサポート(Go 1.11+)では、単純にインポートするだけです:

import "github.com/go-ego/gse"

それ以外の場合、gse パッケージをインストールするには以下のコマンドを実行します:

go get -u github.com/go-ego/gse

使用方法

package main

import (
	_ "embed"
	"fmt"

	"github.com/go-ego/gse"
)

//go:embed testdata/test_en2.txt
var testDict string

//go:embed testdata/test_en.txt
var testEn string

var (
	text  = "To be or not to be, that's the question!"
	test1 = "Hiworld, Helloworld!"
)

func main() {
	// 元の大文字を保持
	// gse.ToLower = false

	var seg1 gse.Segmenter
	seg1.DictSep = ","
	err := seg1.LoadDict("./testdata/test_en.txt")
	if err != nil {
		fmt.Println("Load dictionary error: ", err)
	}

	s1 := seg1.Cut(text)
	fmt.Println("seg1 Cut: ", s1)
	// seg1 Cut:  [to be   or   not to be ,   that's the question!]

	var seg2 gse.Segmenter
	seg2.AlphaNum = true
	seg2.LoadDict("./testdata/test_en_dict3.txt")

	s2 := seg2.Cut(test1)
	fmt.Println("seg2 Cut: ", s2)
	// seg2 Cut:  [hi world ,   hello world !]

	var seg3 gse.Segmenter
	seg3.AlphaNum = true
	seg3.DictSep = ","
	err = seg3.LoadDictEmbed(testDict + "\n" + testEn)
	if err != nil {
		fmt.Println("loadDictEmbed error: ", err)
	}
	s3 := seg3.Cut(text + test1)
	fmt.Println("seg3 Cut: ", s3)
	// seg3 Cut:  [to be   or   not to be ,   that's the question! hi world ,   hello world !]

	// example2()
}

例2:

package main

import (
	"fmt"
	"regexp"

	"github.com/go-ego/gse"
	"github.com/go-ego/gse/hmm/pos"
)

var (
	text = "Hello world, Helloworld. Winter is coming! こんにちは世界, 你好世界."

	new, _ = gse.New("zh,testdata/test_en_dict3.txt", "alpha")

	seg gse.Segmenter
	posSeg pos.Segmenter
)

func main() {
	// デフォルト辞書の読み込み
	seg.LoadDict()
	// 埋め込みデフォルト辞書の読み込み
	// seg.LoadDictEmbed()
	//
	// 簡体字中国語辞書の読み込み
	// seg.LoadDict("zh_s")
	// seg.LoadDictEmbed("zh_s")
	//
	// 繁体字中国語辞書の読み込み
	// seg.LoadDict("zh_t")
	//
	// 日本語辞書の読み込み
	// seg.LoadDict("jp")
	//
	// 辞書の読み込み
	// seg.LoadDict("your gopath"+"/src/github.com/go-ego/gse/data/dict/dictionary.txt")

	cut()

	segCut()
}

func cut() {
	hmm := new.Cut(text, true)
	fmt.Println("cut use hmm: ", hmm)

	hmm = new.CutSearch(text, true)
	fmt.Println("cut search use hmm: ", hmm)
	fmt.Println("analyze: ", new.Analyze(hmm, text))

	hmm = new.CutAll(text)
	fmt.Println("cut all: ", hmm)

	reg := regexp.MustCompile(`(\d+年|\d+月|\d+日|[\p{Latin}]+|[\p{Hangul}]+|\d+\.\d+|[a-zA-Z0-9]+)`)
	text1 := `헬로월드 헬로 서울, 2021年09月10日, 3.14`
	hmm = seg.CutDAG(text1, reg)
	fmt.Println("Cut with hmm and regexp: ", hmm, hmm[0], hmm[6])
}

func analyzeAndTrim(cut []string) {
	a := seg.Analyze(cut, "")
	fmt.Println("analyze the segment: ", a)

	cut = seg.Trim(cut)
	fmt.Println("cut all: ", cut)

	fmt.Println(seg.String(text, true))
	fmt.Println(seg.Slice(text, true))
}

func cutPos() {
	po := seg.Pos(text, true)
	fmt.Println("pos: ", po)
	po = seg.TrimPos(po)
	fmt.Println("trim pos: ", po)

	pos.WithGse(seg)
	po = posSeg.Cut(text, true)
	fmt.Println("pos: ", po)

	po = posSeg.TrimWithPos(po, "zg")
	fmt.Println("trim pos: ", po)
}

func segCut() {
	// テキスト分割
	tb := []byte(text)
	fmt.Println(seg.String(text, true))

	segments := seg.Segment(tb)
	// 分かち書き結果の処理、検索モード
	fmt.Println(gse.ToString(segments, true))
}

カスタム辞書の例を見る

package main

import (
	"fmt"
	_ "embed"

	"github.com/go-ego/gse"
)

//go:embed test_en_dict3.txt
var testDict string

func main() {
	// var seg gse.Segmenter
	// seg.LoadDict("zh, testdata/zh/test_dict.txt, testdata/zh/test_dict1.txt")
	// seg.LoadStop()
	seg, err := gse.NewEmbed("zh, word 20 n"+testDict, "en")
	// seg.LoadDictEmbed()
	seg.LoadStopEmbed()

	text1 := "Hello world, こんにちは世界, 你好世界!"
	s1 := seg.Cut(text1, true)
	fmt.Println(s1)
	fmt.Println("trim: ", seg.Trim(s1))
	fmt.Println("stop: ", seg.Stop(s1))
	fmt.Println(seg.String(text1, true))

	segments := seg.Segment([]byte(text1))
	fmt.Println(gse.ToString(segments))
}

中国語の例を見る

日本語の例を見る

Elasticsearch

elasticsearch と一緒に使用するには?

go-gse-elastic

作者

ライセンス

Gse は主に「Apache License (Version 2.0)」の条項の下で配布されています。 LICENSE-APACHELICENSE を参照してください。

segojiebajiebago)に感謝します。