tree

type DecisionTreeClassifier struct {
	// contains filtered or unexported fields
}

type DecisionTreeClassifierOption func(*DecisionTreeClassifier)

type TreeNode struct {
	IsLeaf       bool      // Whether this is a leaf node
	Feature      int       // Feature index for split (internal nodes)
	Threshold    float64   // Threshold value for split (internal nodes)
	Left         *TreeNode // Left child (values <= threshold)
	Right        *TreeNode // Right child (values > threshold)
	Value        float64   // Predicted value (leaf nodes - regression)
	ClassCounts  []int     // Class counts (leaf nodes - classification)
	PredictClass int       // Predicted class (leaf nodes - classification)
	Impurity     float64   // Node impurity
	NSamples     int       // Number of samples at this node
	Depth        int       // Depth of this node in the tree
}