wordrank.go

package main
import (
	"fmt"
	"os"
	"io/ioutil"
	"strconv"
	"flag"
	"github.com/DavidBelicza/TextRank"
)
func main() {
	file := flag.String("file", "", "file to read")
	flag.Parse()
	osfile, err := os.Open(*file)
	if err != nil { panic(err) }
	data, err := ioutil.ReadAll(osfile)
	if err != nil { panic(err) }

	tr := textrank.NewTextRank()
	rule := textrank.NewDefaultRule()
	language := textrank.NewDefaultLanguage()
	algorithmDef := textrank.NewDefaultAlgorithm()
	tr.Populate(string(data), language, rule)
	tr.Ranking(algorithmDef)
	// rankedPhrases := textrank.FindPhrases(tr)
	rankedwords := textrank.FindSingleWords(tr)
  //  for n, _ := range rankedPhrases {
  //  	fmt.Println(rankedPhrases[n].Right, rankedPhrases[n].Left, rankedPhrases[n].Qty, rankedPhrases[n].Weight)
  //  }
	for n, _ := range rankedwords {
		// fmt.Println(rankedwords[n].Weight)
		strwgt := strconv.FormatFloat(float64(rankedwords[n].Weight), 'f', 3, 64)
		fmt.Println(strwgt + " $:$ " + rankedwords[n].Word)
	}
	
}