Fix/clip PLS

open-spaced-repetition · Dec 16, 2024 · 98720c5 · 98720c5
1 parent 20a712d
commit 98720c5
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/parameters.go b/parameters.go
@@ -101,10 +101,11 @@ func (p *Parameters) nextRecallStability(d float64, s float64, r float64, rating
 }
 
 func (p *Parameters) nextForgetStability(d float64, s float64, r float64) float64 {
-	return p.W[11] *
-		math.Pow(d, -p.W[12]) *
-		(math.Pow(s+1, p.W[13]) - 1) *
-		math.Exp((1-r)*p.W[14])
+	newSMin := s / math.Exp(p.W[17]*p.W[18])
+	return math.Min(newSMin, p.W[11]*
+		math.Pow(d, -p.W[12])*
+		(math.Pow(s+1, p.W[13])-1)*
+		math.Exp((1-r)*p.W[14]))
 }
 
 type FuzzRange struct {