SHORT_TERM_FACTOR

open-spaced-repetition · Oct 11, 2024 · 7145bbe · 7145bbe
1 parent 4023795
commit 7145bbe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -66,7 +66,7 @@
 ]
 
 S_MIN = 0.01
-
+SHORT_TERM_FACTOR = 0.33
 
 class FSRS(nn.Module):
     def __init__(self, w: List[float], float_delta_t: bool = False):
@@ -81,7 +81,7 @@ def stability_after_success(
         easy_bonus = torch.where(rating == 4, self.w[16], 1)
         new_s = state[:, 0] * (
             1
-            + torch.exp(torch.where(delta_t >= 0.66, self.w[8], self.w[19]))
+            + torch.exp(torch.where(delta_t >= SHORT_TERM_FACTOR, self.w[8], self.w[19]))
             * (11 - state[:, 1])
             * torch.pow(state[:, 0], -self.w[9])
             * (torch.exp((1 - r) * self.w[10]) - 1)
@@ -91,7 +91,7 @@ def stability_after_success(
 
     def stability_after_failure(self, state: Tensor, r: Tensor, delta_t: Tensor) -> Tensor:
         new_s = (
-            torch.where(delta_t >= 0.66, self.w[11], self.w[20])
+            torch.where(delta_t >= SHORT_TERM_FACTOR, self.w[11], self.w[20])
             * torch.pow(state[:, 1], -self.w[12])
             * (torch.pow(state[:, 0] + 1, self.w[13]) - 1)
             * torch.exp((1 - r) * self.w[14])