cart

type NodeValue struct {
	// SplitV define the split value.
	SplitV interface{}

	// Class of leaf node.
	Class string

	// SplitAttrName define the name of attribute which cause the split.
	SplitAttrName string

	// Size define number of sample that this node hold before splitting.
	Size int

	// SplitAttrIdx define the attribute which cause the split.
	SplitAttrIdx int

	// IsLeaf define whether node is a leaf or not.
	IsLeaf bool

	// IsContinu define whether the node split is continuous or discrete.
	IsContinu bool
}

type Runtime struct {
	// Tree in classification.
	Tree binary.Tree

	// SplitMethod define the criteria to used for splitting.
	SplitMethod string `json:"SplitMethod"`

	// NRandomFeature if less or equal to zero compute gain on all feature,
	// otherwise select n random feature and compute gain only on selected
	// features.
	NRandomFeature int `json:"NRandomFeature"`

	// OOBErrVal is the last out-of-bag error value in the tree.
	OOBErrVal float64
}