cart

Documentation

Index

• func Gini(y []float64) float64
• type CART
  • func LoadModel(filename string) (*CART, error)
  • func NewCART() *CART
  • func (cart *CART) Fit(X [][]float64, y []float64)
  • func (cart *CART) Predict(X [][]float64) []float64
  • func (cart *CART) SaveModel(filename string) error
• type DecisionNode

Functions

func Gini

func Gini(y []float64) float64

Types

type CART

type CART struct {
	Root *DecisionNode
}

func LoadModel

func LoadModel(filename string) (*CART, error)

func NewCART

func NewCART() *CART

func (*CART) Fit

func (cart *CART) Fit(X [][]float64, y []float64)

func (*CART) Predict

func (cart *CART) Predict(X [][]float64) []float64

func (*CART) SaveModel

func (cart *CART) SaveModel(filename string) error

type DecisionNode

type DecisionNode struct {
	FeatureIndex int
	Threshold    float64
	Left         *DecisionNode
	Right        *DecisionNode
	Value        float64
	IsLeaf       bool
}