ensemble

Documentation

Index

• type RandomForest
  • func NewRandomForest(forestSize int, features int) *RandomForest
  • func (f *RandomForest) Fit(on *base.Instances)
  • func (f *RandomForest) Predict(with *base.Instances) *base.Instances
  • 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

NewRandomForests 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.Instances)

Train builds the RandomForest on the specified instances

func (*RandomForest) Predict

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

Predict generates predictions from a trained RandomForest

func (*RandomForest) String

func (f *RandomForest) String() string