ensemble

Documentation

Index

• type RandomForest
  • func NewRandomForest(forestSize int, features int) *RandomForest
  • func (f *RandomForest) Fit(on base.FixedDataGrid)
  • func (f *RandomForest) Predict(with base.FixedDataGrid) base.FixedDataGrid
  • func (f *RandomForest) String() string

Types

type RandomForest

type RandomForest struct {
	base.BaseClassifier
	ForestSize int
	Features   int
	Model      *meta.BaggedModel
}

RandomForest classifies instances using an ensemble of bagged random decision trees.

func NewRandomForest

func NewRandomForest(forestSize int, features int) *RandomForest

NewRandomForest generates and return a new random forests forestSize controls the number of trees that get built features controls the number of features used to build each tree.

func (*RandomForest) Fit

func (f *RandomForest) Fit(on base.FixedDataGrid)

Fit builds the RandomForest on the specified instances

func (*RandomForest) Predict

func (f *RandomForest) Predict(with base.FixedDataGrid) base.FixedDataGrid

Predict generates predictions from a trained RandomForest.

func (*RandomForest) String

func (f *RandomForest) String() string

String returns a human-readable representation of this tree.