linreg

Documentation

Overview

Package linreg provides a basic implementation of linear regression with gradient descent on two dimensional data.

Index

• func Gradient(xs, ys []float64, m, c float64) (cost, dm, dc float64)
• func LinearRegression(xs, ys []float64, iterations int, alpha float64) (m, c float64)

Functions

func Gradient

func Gradient(xs, ys []float64, m, c float64) (cost, dm, dc float64)

Gradient computes the cost function and its gradients.

func LinearRegression

func LinearRegression(xs, ys []float64, iterations int, alpha float64) (m, c float64)

LinearRegression runs the requested number of iterations of gradient descent and returns the latest approximated coefficients.