tldr is a golang package to summarize a text automatically using lexrank algorithm.
There are two main steps in lexrank, weighing, and ranking. tldr have two weighing and two ranking algorithm included, they are Jaccard coeficient and Hamming distance, then PageRank and centrality, respectively. The default settings use Hamming distance and pagerank.
$ go test -bench . -benchmem -benchtime 5s -cpu 4
Running Suite: Tldr Suite
=========================
Random Seed: 1759506557
Will run 8 of 8 specs
••••••••
Ran 8 of 8 Specs in 0.012 seconds
SUCCESS! -- 8 Passed | 0 Failed | 0 Pending | 0 Skipped
goos: linux
goarch: amd64
pkg: github.com/didasy/tldr
cpu: AMD Ryzen 5 5600G with Radeon Graphics
BenchmarkSummarizeCentralityHamming-4 5877 896338 ns/op 177320 B/op 1898 allocs/op
BenchmarkSummarizeCentralityJaccard-4 6562 885374 ns/op 177221 B/op 1898 allocs/op
BenchmarkSummarizePagerankHamming-4 5832 962000 ns/op 200830 B/op 2086 allocs/op
BenchmarkSummarizePagerankJaccard-4 5949 962579 ns/op 200865 B/op 2086 allocs/op
PASS
ok github.com/didasy/tldr 22.840s
So, not bad huh?
go get github.com/didasy/tldr
package main
import (
"fmt"
"io/ioutil"
"github.com/didasy/tldr"
)
func main() {
intoSentences := 3
textB, _ := ioutil.ReadFile("./sample.txt")
text := string(textB)
bag := tldr.New()
result, _ := bag.Summarize(text, intoSentences)
fmt.Println(result)
}
To test, just run go test, but you need to have gomega and ginkgo installed.
tldr depends on pagerank package, and you can install it with go get github.com/alixaxel/pagerank.
Check the LICENSE file. tldr: MIT.