Update fsrs_optimizer.py

user1823 · Oct 17, 2024 · c9517e8 · c9517e8
1 parent b2d13cc
commit c9517e8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -107,7 +107,10 @@ def init_d(self, rating: Tensor) -> Tensor:
 
     def next_d(self, state: Tensor, rating: Tensor) -> Tensor:
         delta_d = - self.w[6] * (rating - 3)
-        new_d = state[:, 1] + delta_d * (1 - (state[:, 1]/10)) # linear dampening
+        if delta_d > 0:
+            new_d = state[:, 1] + delta_d * (1 - (state[:, 1]/10))
+        if delta_d < 0:
+            new_d = state[:, 1] + delta_d * (1 - ((11 - state[:, 1])/10))
         new_d = self.mean_reversion(self.init_d(4), new_d)
         return new_d