Fix/set sigma to 1 / sqrt(total_cnt) (#103)

open-spaced-repetition · Apr 7, 2024 · 65a4bf5 · 65a4bf5
1 parent a7504d0
commit 65a4bf5
Show file tree

Hide file tree

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

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -714,7 +714,7 @@ def cal_stability(group: pd.DataFrame) -> pd.DataFrame:
                         power_forgetting_curve,
                         group["delta_t"],
                         group["retention"],
-                        sigma=1 / group["total_cnt"],
+                        sigma=1 / np.sqrt(group["total_cnt"]),
                     )[0][0],
                     1,
                 )