From c7423f8f727f53aee1550ee281fd432a823337e0 Mon Sep 17 00:00:00 2001
From: Bel LaPointe <breel@qualtrics.com>
Date: Thu, 26 Oct 2023 06:16:23 -0600
Subject: [PATCH] median of medians

---
 ledger/predict.go | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/ledger/predict.go b/ledger/predict.go
index 5bffcf6..38c1bc2 100644
--- a/ledger/predict.go
+++ b/ledger/predict.go
@@ -80,17 +80,29 @@ func registerWithContributionPredictionForNameForCurrency(reg Register, window t
 		if wouldBe == 0 {
 			wouldBe = 2
 		}
-		return contributions[len(contributions)-wouldBe:]
+		return contributions[len(contributions)-1-wouldBe:]
 	}
 
 	eighth := contributsSlice(7.0 / 8.0)
 	quarter := contributsSlice(3.0 / 4.0)
 	half := contributsSlice(1.0 / 2.0)
 	medianValueDelta := func() float64 {
-		return (4.0*getMedianValueDelta(eighth) + 2.0*getMedianValueDelta(quarter) + 1.0*getMedianValueDelta(half)) / (4.0 + 2.0 + 1.0)
+		medians := []float64{
+			getMedianValueDelta(eighth),
+			getMedianValueDelta(quarter),
+			getMedianValueDelta(half),
+		}
+		slices.Sort(medians)
+		return medians[1]
 	}()
 	medianLapse := func() time.Duration {
-		return (4.0*getMedianLapse(eighth) + 2.0*getMedianLapse(quarter) + 1.0*getMedianLapse(half)) / (4.0 + 2.0 + 1.0)
+		medians := []time.Duration{
+			getMedianLapse(eighth),
+			getMedianLapse(quarter),
+			getMedianLapse(half),
+		}
+		slices.Sort(medians)
+		return medians[1]
 	}()
 
 	latest := func() float64 {