inference

Documentation

Index

• type Ensemble
  • func (e *Ensemble) Name() string
  • func (e *Ensemble) Predict(features mat.SparseMatrix) (mat.Matrix, error)
  • func (e *Ensemble) PredictProba(features mat.SparseMatrix) (mat.Matrix, error)
  • func (e *Ensemble) PredictRegression(features mat.SparseMatrix, baseVal float32) (mat.Matrix, error)
• type EnsembleBase

Types

type Ensemble

type Ensemble struct {
	EnsembleBase
	activation.Activation
}

Ensemble struct contains ensemble model interface that a model needs to implement.

func (*Ensemble) Name

func (e *Ensemble) Name() string

Name returns ensemble model name.

func (*Ensemble) Predict

func (e *Ensemble) Predict(features mat.SparseMatrix) (mat.Matrix, error)

Predict predicts class using ensemble model interface. If model is a binary classification model, the prediction results will be probabilities instead of classes.

func (*Ensemble) PredictProba

func (e *Ensemble) PredictProba(features mat.SparseMatrix) (mat.Matrix, error)

PredictProba predicts probabilities using ensemble model interface.

func (*Ensemble) PredictRegression

func (e *Ensemble) PredictRegression(features mat.SparseMatrix, baseVal float32) (mat.Matrix, error)

PredictRegression predicts float number for regression task using ensemble model interface.

type EnsembleBase

type EnsembleBase interface {
	PredictInner(features mat.SparseVector) (mat.Vector, error)
	Name() string
	NumClasses() int
}

EnsembleBase contains interface of a base model.