Feat/new initial values of S0 from benchmark (#49)

open-spaced-repetition · Oct 14, 2023 · 28173a1 · 28173a1
1 parent fa63f61
commit 28173a1
Show file tree

Hide file tree

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

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -787,9 +787,9 @@ def define_model(self):
         """Step 3"""
         self.init_w = [
             0.4,
-            0.6,
-            2.4,
-            5.8,
+            0.9,
+            2.3,
+            10.9,
             4.93,
             0.94,
             0.86,
@@ -830,8 +830,7 @@ def pretrain(self, dataset=None, verbose=True):
         rating_count = {}
         average_recall = self.dataset["y"].mean()
         plots = []
-        s0_size = self.S0_dataset_group.shape[0]
-        r_s0_default = {"1": 0.4, "2": 0.6, "3": 2.4, "4": 5.8}
+        r_s0_default = {"1": 0.4, "2": 0.9, "3": 2.3, "4": 10.9}
 
         for first_rating in ("1", "2", "3", "4"):
             group = self.S0_dataset_group[