randomforest

type Node struct {
	Tree *Tree

	FeatureIndex int
	Value        float64

	IdxCnt int

	Left  *Node
	Right *Node

	Label int
}

type RandomForest struct {
	TreeCount    int            // Number of trees to include in forest
	MaxDepth     int            //
	MinSize      int            //
	FeatureCount int            // Number of features to include in each tree
	Trees        []*Tree        // List of trees created in `Fit`
	Evaluator    SplitEvaluator // Function to evaluate each split
	SampleSize   float64        // Ratio of rows to consider in each tree

	Data   [][]float64
	Labels []string
}

type SplitEvaluator func(*Tree, []int, []int) float64

type Tree struct {
	Forest       *RandomForest
	Root         *Node
	Data         [][]float64
	Labels       []int
	UniqueCounts []float64

	MaxDepth int
	// contains filtered or unexported fields
}