func main() {
defer un(trace("smote"))
flag.Parse()
if len(flag.Args()) <= 0 {
usage()
os.Exit(1)
}
fcfg := flag.Arg(0)
// Parsing config file and parameter.
smote, e := createSmote(fcfg)
if e != nil {
panic(e)
}
// Get dataset.
dataset := tabula.Claset{}
_, e = dsv.SimpleRead(fcfg, &dataset)
if e != nil {
panic(e)
}
fmt.Println("[smote] Dataset:", &dataset)
row := dataset.GetRow(0)
fmt.Println("[smote] sample:", row)
e = runSmote(smote, &dataset)
if e != nil {
panic(e)
}
if !merge {
return
}
e = runMerge(smote, &dataset)
if e != nil {
panic(e)
}
}
func test() {
testset := tabula.Claset{}
_, e := dsv.SimpleRead(testCfg, &testset)
if e != nil {
panic(e)
}
fmt.Println(tag, "Test set:", &testset)
fmt.Println(tag, "Sample test set:", testset.GetRow(0))
predicts, cm, probs := crforest.ClassifySetByWeight(&testset, nil)
fmt.Println("[crf] Test set CM:", cm)
crforest.Performance(&testset, predicts, probs)
e = crforest.WritePerformance()
if e != nil {
panic(e)
}
}