ml

package

v0.0.4 Latest Latest Go to latest Published: Oct 10, 2020 License: MIT Imports: 5 Imported by: 0

Details

View Source

const (
	False = float64(0)
	True  = float64(1)
)

Predefined values for true and false.

This section is empty.

func Bool(t bool) float64

Bool converts a bool to a value representing false or true.

func Normalize(xs *mat.Dense) error

Normalize normalizes the the given feature vectors.

type LR struct {
	LearningRate float64
	Ntrain       int
	// contains filtered or unexported fields
}

LR implements LinearRegression

func (lr *LR) Fit(x *mat.Dense, y *mat.VecDense)

Fit fits the linear regression model.

func (lr *LR) MarshalJSON() ([]byte, error)

MarshalJSON implements the json.Marshal interface.

func (lr *LR) Predict(x *mat.Dense, t float64) *mat.VecDense

Predict calculates the predictions for the given values.

func (lr *LR) PredictProb(x *mat.Dense) *mat.VecDense

PredictProb calculates the probablility predictions for the given values.

func (lr *LR) UnmarshalJSON(data []byte) error

UnmarshalJSON implements the json.Unmarshal interface.