forests get Type, so Predict can be generic.

applyforest/applyforest.go

@@ -24,7 +24,7 @@ func main() {
 	var sum bool
 	flag.BoolVar(&sum, "sum", false, "Force numeric sum voting (for gradient boosting etc).")
 	var expit bool
-	flag.BoolVar(&expit, "expit", false, "Expit (inverst logit) transform data (for gradient boosting classification).")
+	flag.BoolVar(&expit, "expit", false, "Expit (inverse logit) transform data (for gradient boosting classification).")
 	var cat bool
 	flag.BoolVar(&cat, "mode", false, "Force categorical (mode) voting.")
 
@@ -46,6 +46,10 @@ func main() {
 	if err != nil {
 		log.Fatal(err)
 	}
+	if forest.PredCfg.Gradboost != 0 {
+		sum = true
+	}
+	fmt.Printf("Forest is of type %v.\n", forest.Type.String())
 
 	var predfile *os.File
 	if *predfn != "" {

catballotbox.go

@@ -126,3 +126,9 @@ func (bb *CatBallotBox) TallyError(feature Feature) float64 {
 	e = 1.0 - e
 	return e
 }
+
+// TallyNum always returns NaN for a SumBallotBox; we
+// meet the demands of the expanded VoteTallyer interface.
+func (bb *CatBallotBox) TallyNum(i int) (predicted float64) {
+	return nan
+}

cfpred/cfpred.go

@@ -0,0 +1,73 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/lytics/CloudForest"
+)
+
+func main() {
+	fm := flag.String("fm",
+		"featurematrix.afm", "AFM formated feature matrix containing data.")
+	rf := flag.String("rfpred",
+		"rface.sf", "A predictor forest.")
+	predfn := flag.String("preds",
+		"", "The name of a file to write the predictions into.")
+
+	flag.Parse()
+
+	//Parse Data
+	data, err := CloudForest.LoadAFM(*fm)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	forestfile, err := os.Open(*rf) // For read access.
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer forestfile.Close()
+	forestreader := CloudForest.NewForestReader(forestfile)
+	forest, err := forestreader.ReadForest()
+	if err != nil {
+		log.Fatal(err)
+	}
+	fmt.Printf("Forest is of type %v.\n", forest.Type.String())
+
+	var predfile *os.File
+	if *predfn != "" {
+		predfile, err = os.Create(*predfn)
+		if err != nil {
+			log.Fatal(err)
+		}
+		defer predfile.Close()
+	}
+	targeti, hasTarget := data.Map[forest.Target]
+
+	preds, err := CloudForest.Predict(data, forest)
+	if err != nil {
+		log.Fatal(err)
+	}
+	if *predfn != "" {
+		fmt.Printf("Outputting label predicted actual tsv to %v\n", *predfn)
+		for i, l := range data.CaseLabels {
+			actual := "NA"
+			if hasTarget {
+				actual = data.Data[targeti].GetStr(i)
+			}
+
+			var result string
+			if preds.IsReal {
+				result = fmt.Sprintf("%v", preds.Real[i])
+			} else {
+				result = preds.Cat[i]
+			}
+
+			fmt.Fprintf(predfile, "%v\t%v\t%v\n", l, result, actual)
+		}
+	}
+
+}