Feat/get retrievability

open-spaced-repetition · Sep 12, 2024 · 8babacb · 8babacb
1 parent 3c34707
commit 8babacb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/fsrs.go b/fsrs.go
@@ -19,3 +19,13 @@ func (f *FSRS) Repeat(card Card, now time.Time) RecordLog {
 func (f *FSRS) Next(card Card, now time.Time, grade Rating) SchedulingInfo {
 	return f.scheduler(card, now).Review(grade)
 }
+
+func (f *FSRS) GetRetrievability(card Card, now time.Time) float64 {
+	if card.State == New {
+		return 0
+	} else {
+		elapsedDays := now.Sub(card.LastReview).Hours() / 24
+		retrievability := f.Parameters.forgettingCurve(elapsedDays, card.Stability)
+		return retrievability
+	}
+}