Ejemplo n.º 1
0
func (m *LinearModel) FitTransform(x, y num.Matrix) num.Matrix {
	xT := x.Transpose()
	m.Coefs = (xT.Times(x)).Inverse().Times(xT).Times(y)
	predicted := x.Times(m.Coefs)
	m.Error = 0.0
	for j := 0; j < predicted.Shape[0]; j++ {
		m.Error += math.Pow(predicted.Data[j][0]-y.Data[j][0], 2)
	}
	m.Error /= float64(predicted.Shape[0])

	return predicted
}
Ejemplo n.º 2
0
func (m LinearModel) Transform(x num.Matrix) num.Matrix {
	return x.Times(m.Coefs)
}