linear

Documentation

Index

• Variables
• func Min(values []float64) (min float64, err error)
• type LR
  • func LinearRegression(train core.Train, targetColumn int, method LRtype) *LR
  • func (lr *LR) DeterminationCoefficient(testTrain core.Train) (coef float64)
  • func (lr *LR) Fit() (err error)
  • func (lr *LR) Predict(testTrainR core.Train, testTrainM *mat.Dense) (predictResult *mat.VecDense)
  • func (lr *LR) PredictDense(testTrain *mat.Dense) (predictResult *mat.VecDense)
  • func (lr *LR) PredictTrain(testTrain core.Train) (predictResult *mat.VecDense)
• type LRtype

Variables

var (
	SVDDecompositionError  = errors.New("ошибка во время SVD разложения")
	SVDResultComputeError  = errors.New("вектор b посчитан неправильно")
	CoefBeforeFittingError = errors.New("обучить")
)

Functions

func Min

func Min(values []float64) (min float64, err error)

Types

type LR

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

LinearRegression is main struct for linear regression package

func LinearRegression

func LinearRegression(train core.Train, targetColumn int, method LRtype) *LR

LinearRegression prepare structure and data for fitting

func (*LR) DeterminationCoefficient

func (lr *LR) DeterminationCoefficient(testTrain core.Train) (coef float64)

DeterminationCoefficient return coefficient of determination

func (*LR) Fit

func (lr *LR) Fit() (err error)

Fit train to LR

func (*LR) Predict

func (lr *LR) Predict(testTrainR core.Train, testTrainM *mat.Dense) (predictResult *mat.VecDense)

Predict wrapper for PredictTrain and PredictDense

func (*LR) PredictDense

func (lr *LR) PredictDense(testTrain *mat.Dense) (predictResult *mat.VecDense)

Predict does predict by Dense

func (*LR) PredictTrain

func (lr *LR) PredictTrain(testTrain core.Train) (predictResult *mat.VecDense)

Predict does predict by Train

type LRtype

type LRtype int

LRtype specifies the treatment of solving

const (
	// QR specifies solving with QR decomposition
	QR LRtype = 1 << (iota + 1)
	// SVD specifies solving with SVD decomposition
	SVD
)