ols

package module

v0.0.0-...-0dd20d7 Latest Latest Go to latest Published: Oct 21, 2016 License: MIT Imports: 3 Imported by: 0

Details

This section is empty.

This section is empty.

func SetLogger(logger *logrus.Logger)

SetLogger replaces the default logger

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

Model handles the regression

func NewModel(x, y mat64.Matrix) *Model

NewModel creates a new model with an intercept by default

func NewModelWithIntercept(x, y mat64.Matrix, intercept bool) *Model

NewModelWithIntercept creates a new OLS model with or without an intercept x or y can be nil, just needs to be updated before training

func (m *Model) Dims() (int, int)

Dims returns number of independent variabes and count of rows

func (m *Model) SetX(data mat64.Matrix)

SetX replaces the independent variables in the model

func (m *Model) SetY(y mat64.Matrix)

SetY replaces the dependent model NOTE: only uses first column of the matrix

func (m *Model) Train() *mat64.Dense

Train - process the data and return the result