|
@@ -1,7 +1,7 @@
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
- // "encoding/json"
|
|
|
+ "encoding/json"
|
|
|
"fmt"
|
|
|
"io"
|
|
|
"log"
|
|
@@ -39,24 +39,37 @@ func saveBayesToFile() {
|
|
|
|
|
|
dumpfile := os.Getenv("DUMPFILE")
|
|
|
if dumpfile == "" {
|
|
|
- dumpfile = "bayes.json"
|
|
|
+ dumpfile = "./"
|
|
|
}
|
|
|
|
|
|
- err := ZClassifier.Learning.bayez.WriteClassesToFile(dumpfile)
|
|
|
+ var statsREPORT string
|
|
|
|
|
|
+ statsREPORT = "\n"
|
|
|
+
|
|
|
+ ZClassifier.STATS.Range(func(key interface{}, value interface{}) bool {
|
|
|
+
|
|
|
+ statsREPORT += fmt.Sprintf("{\"%s\" : \"%d\"}\n", key.(string), value.(int64))
|
|
|
+ return true
|
|
|
+ })
|
|
|
+
|
|
|
+ writeToFile(dumpfile+"stats.json", statsREPORT)
|
|
|
+
|
|
|
+ badScores, err := json.MarshalIndent(ZClassifier.Learning.bayez.datas[Bad], "", " ")
|
|
|
if err != nil {
|
|
|
- log.Println("Error Writing file: ", dumpfile, err.Error())
|
|
|
+ badScores = []byte(err.Error())
|
|
|
}
|
|
|
|
|
|
- // var statsREPORT string
|
|
|
+ writeToFile(dumpfile+"BAD.json", string(badScores))
|
|
|
+ log.Println(string(badScores))
|
|
|
|
|
|
- // statsREPORT = "\n"
|
|
|
+ goodScores, err := json.MarshalIndent(ZClassifier.Learning.bayez.datas[Good], "", " ")
|
|
|
+ if err != nil {
|
|
|
+ goodScores = []byte(err.Error())
|
|
|
+ }
|
|
|
|
|
|
- // Classifier.STATS.Range(func(key interface{}, value interface{}) bool {
|
|
|
+ writeToFile(dumpfile+"GOOD.json", string(goodScores))
|
|
|
|
|
|
- // statsREPORT += fmt.Sprintf("{\"%s\" : \"%d\"}\n", key.(string), value.(int64))
|
|
|
- // return true
|
|
|
- // })
|
|
|
+ log.Println(string(goodScores))
|
|
|
|
|
|
// dumpfile := os.Getenv("DUMPFILE")
|
|
|
// if dumpfile == "" {
|