From 65a4bf50d12ec9bd518a4280d7b19d6b29245452 Mon Sep 17 00:00:00 2001
From: Jarrett Ye <jarrett.ye@outlook.com>
Date: Sun, 7 Apr 2024 14:46:09 +0800
Subject: [PATCH] Fix/set sigma to 1 / sqrt(total_cnt) (#103)

---
 pyproject.toml                       | 2 +-
 src/fsrs_optimizer/fsrs_optimizer.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index e09b485..46f7532 100644
--- 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
index 6201ec9..97a37ec 100644
--- 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,
                 )