Fine tuned w1 and w2 for pretraining (#158)

open-spaced-repetition · Jan 6, 2025 · 24dfb16 · 24dfb16
1 parent 333651f
commit 24dfb16
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "FSRS-Optimizer"
-version = "5.6.3"
+version = "5.6.4"
 readme = "README.md"
 dependencies = [
     "matplotlib>=3.7.0",

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -1104,8 +1104,8 @@ def loss(stability):
                     else:
                         rating_stability[small_rating] = rating_stability[big_rating]
 
-        w1 = 3 / 5
-        w2 = 3 / 5
+        w1 = 0.41
+        w2 = 0.54
 
         if len(rating_stability) == 0:
             raise Exception("Not enough data for pretraining!")