fixing handling of offset and window
This commit is contained in:
parent
6533e951e7
commit
fc0f274f51
1 changed files with 3 additions and 2 deletions
|
@ -40,7 +40,7 @@ func (lrs LinearRegressionSeries) Len() int {
|
||||||
// GetWindow returns the window size.
|
// GetWindow returns the window size.
|
||||||
func (lrs LinearRegressionSeries) GetWindow() int {
|
func (lrs LinearRegressionSeries) GetWindow() int {
|
||||||
if lrs.Window == 0 {
|
if lrs.Window == 0 {
|
||||||
return lrs.InnerSeries.Len()
|
return lrs.InnerSeries.Len() - lrs.GetOffset()
|
||||||
}
|
}
|
||||||
return lrs.Window
|
return lrs.Window
|
||||||
}
|
}
|
||||||
|
@ -69,7 +69,8 @@ func (lrs *LinearRegressionSeries) GetValue(index int) (x, y float64) {
|
||||||
lrs.computeCoefficients()
|
lrs.computeCoefficients()
|
||||||
}
|
}
|
||||||
offset := lrs.GetOffset()
|
offset := lrs.GetOffset()
|
||||||
x, y = lrs.InnerSeries.GetValue(index + offset)
|
effectiveIndex := MinInt(index+offset, lrs.InnerSeries.Len())
|
||||||
|
x, y = lrs.InnerSeries.GetValue(effectiveIndex)
|
||||||
y = (lrs.m * lrs.normalize(x)) + lrs.b
|
y = (lrs.m * lrs.normalize(x)) + lrs.b
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue