algorithm_cart

package

v0.0.0-...-791d115 Latest Latest Go to latest Published: Dec 3, 2021 License: AGPL-3.0 Imports: 3 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type CART struct {
	Scores []float64
	Trees  []*Tree
	// contains filtered or unexported fields
}

func (k *CART) Accuracy(predict []int64, labels []int64) float64

Accuracy calculates the quality metric

func (k *CART) CrossValSplit(data [][]float64) [][][]float64

CrossValSplit divides data into folds

func (k *CART) CrossValidationScore(data [][]float64) float64

CrossValidationScore evaluates the algorithm on different partitions into training and test

func (k *CART) InitAlgorithm(depth int, minCount int, countFold int)

InitAlgorithm shows tree in console

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

Tree - main node

func (t *Tree) Predict(features [][]float64) []int64

Predict makes a number of predictions based on the constructed tree

func (t *Tree) PrintTree()

PrintTree shows tree in console