JSON serialization

[nnqq]

Hi. My edits:

• JSON serialization as an option (gob by default, so it have full backward compatibility). Also JSON have around 25% less file size
• Minor codestyle fixes - error check in defer, spread operator in test
• Go mod file

2020/07/24 20:32:04 gob [One Two Three Four Five Six Seven Eight Nine Ten]
2020/07/24 20:32:04 gob size 816
2020/07/24 20:32:04 json [One Two Three Four Five Six Seven Eight Nine Ten]
2020/07/24 20:32:04 json size 611

package main

import (
	"github.com/jbrukh/bayesian"
	"log"
	"os"
	"path"
)

func write(ser bayesian.Serializer) {
	const (
		One   bayesian.Class = "One"
		Two   bayesian.Class = "Two"
		Three bayesian.Class = "Three"
		Four  bayesian.Class = "Four"
		Five  bayesian.Class = "Five"
		Six   bayesian.Class = "Six"
		Seven bayesian.Class = "Seven"
		Eight bayesian.Class = "Eight"
		Nine  bayesian.Class = "Nine"
		Ten   bayesian.Class = "Ten"
	)

	classifier := bayesian.NewClassifier(One, Two, Three, Four, Five, Six, Seven, Eight, Nine, Ten)
	oneStuff := []string{"lorem", "ipsum", "dolor"}
	twoStuff := []string{"sit", "amet", "consectetur"}
	threeStuff := []string{"adipiscing", "elit", "sed"}
	fourStuff := []string{"do", "eiusmod", "tempor"}
	fiveStuff := []string{"incididunt", "ut", "labore"}
	sixStuff := []string{"et", "dolore", "magna"}
	sevenStuff := []string{"aliqua", "ut", "enim"}
	eightStuff := []string{"ad", "minim", "veniam"}
	nineStuff := []string{"quis", "nostrud", "exercitation"}
	tenStuff := []string{"ullamco", "laboris", "nisi"}

	classifier.Learn(oneStuff, One)
	classifier.Learn(twoStuff, Two)
	classifier.Learn(threeStuff, Three)
	classifier.Learn(fourStuff, Four)
	classifier.Learn(fiveStuff, Five)
	classifier.Learn(sixStuff, Six)
	classifier.Learn(sevenStuff, Seven)
	classifier.Learn(eightStuff, Eight)
	classifier.Learn(nineStuff, Nine)
	classifier.Learn(tenStuff, Ten)

	wd, err := os.Getwd()
	if err != nil {
		panic(err)
	}

	err = classifier.WriteToFile(path.Join(wd, "out_"+string(ser)), ser)
	if err != nil {
		panic(err)
	}
}

func read(ser bayesian.Serializer) {
	wd, err := os.Getwd()
	if err != nil {
		panic(err)
	}

	file := path.Join(wd, "out_"+string(ser))

	classifier, err := bayesian.NewClassifierFromFile(file, ser)
	if err != nil {
		panic(err)
	}

	f, err := os.Open(file)
	if err != nil {
		panic(err)
	}
	info, err := f.Stat()
	if err != nil {
		panic(err)
	}

	log.Println(ser, classifier.Classes)
	log.Println(ser, "size", info.Size())
}

func main() {
	write(bayesian.Gob)
	read(bayesian.Gob)

	write(bayesian.JSON)
	read(bayesian.JSON)
}