go-chart/linear_regression_series.go

148 lines
3.7 KiB
Go
Raw Normal View History

2016-07-27 21:34:15 +02:00
package chart
import (
"fmt"
"github.com/wcharczuk/go-chart/seq"
util "github.com/wcharczuk/go-chart/util"
)
2017-02-03 20:26:53 +01:00
2016-07-27 21:34:15 +02:00
// LinearRegressionSeries is a series that plots the n-nearest neighbors
// linear regression for the values.
type LinearRegressionSeries struct {
Name string
Style Style
2016-08-01 01:54:09 +02:00
YAxis YAxisType
2016-07-27 21:34:15 +02:00
Limit int
2016-07-27 21:34:15 +02:00
Offset int
InnerSeries ValuesProvider
2016-07-27 21:34:15 +02:00
m float64
b float64
avgx float64
stddevx float64
}
// GetName returns the name of the time series.
func (lrs LinearRegressionSeries) GetName() string {
return lrs.Name
}
// GetStyle returns the line style.
func (lrs LinearRegressionSeries) GetStyle() Style {
return lrs.Style
}
// GetYAxis returns which YAxis the series draws on.
2016-08-01 01:54:09 +02:00
func (lrs LinearRegressionSeries) GetYAxis() YAxisType {
2016-07-27 21:34:15 +02:00
return lrs.YAxis
}
// Len returns the number of elements in the series.
func (lrs LinearRegressionSeries) Len() int {
return util.Math.MinInt(lrs.GetLimit(), lrs.InnerSeries.Len()-lrs.GetOffset())
2016-07-27 21:34:15 +02:00
}
// GetLimit returns the window size.
func (lrs LinearRegressionSeries) GetLimit() int {
if lrs.Limit == 0 {
2016-07-27 21:54:40 +02:00
return lrs.InnerSeries.Len()
2016-07-27 21:34:15 +02:00
}
return lrs.Limit
2016-07-27 21:34:15 +02:00
}
// GetEndIndex returns the effective limit end.
2016-07-27 21:54:40 +02:00
func (lrs LinearRegressionSeries) GetEndIndex() int {
2017-04-25 08:00:28 +02:00
windowEnd := lrs.GetOffset() + lrs.GetLimit()
innerSeriesLastIndex := lrs.InnerSeries.Len() - 1
return util.Math.MinInt(windowEnd, innerSeriesLastIndex)
2016-07-27 21:34:15 +02:00
}
// GetOffset returns the data offset.
func (lrs LinearRegressionSeries) GetOffset() int {
if lrs.Offset == 0 {
return 0
}
return lrs.Offset
}
// GetValues gets a value at a given index.
func (lrs *LinearRegressionSeries) GetValues(index int) (x, y float64) {
if lrs.InnerSeries == nil || lrs.InnerSeries.Len() == 0 {
2016-07-27 21:34:15 +02:00
return
}
if lrs.m == 0 && lrs.b == 0 {
lrs.computeCoefficients()
}
offset := lrs.GetOffset()
effectiveIndex := util.Math.MinInt(index+offset, lrs.InnerSeries.Len())
x, y = lrs.InnerSeries.GetValues(effectiveIndex)
2016-07-27 21:34:15 +02:00
y = (lrs.m * lrs.normalize(x)) + lrs.b
return
}
// GetLastValues computes the last linear regression value.
func (lrs *LinearRegressionSeries) GetLastValues() (x, y float64) {
if lrs.InnerSeries == nil || lrs.InnerSeries.Len() == 0 {
2016-07-27 21:34:15 +02:00
return
}
if lrs.m == 0 && lrs.b == 0 {
lrs.computeCoefficients()
}
2016-07-27 21:54:40 +02:00
endIndex := lrs.GetEndIndex()
x, y = lrs.InnerSeries.GetValues(endIndex)
2016-07-27 21:34:15 +02:00
y = (lrs.m * lrs.normalize(x)) + lrs.b
return
}
func (lrs *LinearRegressionSeries) normalize(xvalue float64) float64 {
return (xvalue - lrs.avgx) / lrs.stddevx
}
// computeCoefficients computes the `m` and `b` terms in the linear formula given by `y = mx+b`.
func (lrs *LinearRegressionSeries) computeCoefficients() {
startIndex := lrs.GetOffset()
2016-07-27 21:54:40 +02:00
endIndex := lrs.GetEndIndex()
2016-07-27 21:34:15 +02:00
p := float64(endIndex - startIndex)
xvalues := seq.NewBufferWithCapacity(lrs.Len())
2016-07-27 21:34:15 +02:00
for index := startIndex; index < endIndex; index++ {
x, _ := lrs.InnerSeries.GetValues(index)
2016-07-27 21:34:15 +02:00
xvalues.Enqueue(x)
}
lrs.avgx = seq.Seq{Provider: xvalues}.Average()
lrs.stddevx = seq.Seq{Provider: xvalues}.StdDev()
2016-07-27 21:34:15 +02:00
var sumx, sumy, sumxx, sumxy float64
for index := startIndex; index < endIndex; index++ {
x, y := lrs.InnerSeries.GetValues(index)
2016-07-27 21:34:15 +02:00
x = lrs.normalize(x)
sumx += x
sumy += y
sumxx += x * x
sumxy += x * y
}
lrs.m = (p*sumxy - sumx*sumy) / (p*sumxx - sumx*sumx)
lrs.b = (sumy / p) - (lrs.m * sumx / p)
}
// Render renders the series.
func (lrs *LinearRegressionSeries) Render(r Renderer, canvasBox Box, xrange, yrange Range, defaults Style) {
style := lrs.Style.InheritFrom(defaults)
2016-07-30 01:36:29 +02:00
Draw.LineSeries(r, canvasBox, xrange, yrange, style, lrs)
2016-07-27 21:34:15 +02:00
}
2017-02-03 20:26:53 +01:00
// Validate validates the series.
func (lrs *LinearRegressionSeries) Validate() error {
if lrs.InnerSeries == nil {
return fmt.Errorf("linear regression series requires InnerSeries to be set")
}
return nil
}