Fix/filter threshold (#373)

open-spaced-repetition · Jul 20, 2023 · a6c62ff · a6c62ff
1 parent ea612f1
commit a6c62ff
Show file tree

Hide file tree

Showing 3 changed files with 339 additions and 336 deletions.
diff --git a/fsrs4anki_optimizer.ipynb b/fsrs4anki_optimizer.ipynb
diff --git a/package/fsrs4anki_optimizer/fsrs4anki_optimizer.py b/package/fsrs4anki_optimizer/fsrs4anki_optimizer.py
@@ -407,7 +407,7 @@ def remove_outliers(group: pd.DataFrame) -> pd.DataFrame:
             Q3 = group['delta_t'].quantile(0.75)
             IQR = Q3 - Q1
             threshold = Q3 + 1.5 * IQR
-            group = group[group['delta_t'] < threshold]
+            group = group[group['delta_t'] <= threshold]
             return group
 
         df[df['i'] == 2] = df[df['i'] == 2].groupby(by=['r_history', 't_history'], as_index=False, group_keys=False).apply(remove_outliers)

diff --git a/package/pyproject.toml b/package/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fsrs4anki_optimizer"
-version = "4.2.0"
+version = "4.2.1"
 readme = "README.md"
 dependencies = [
     "matplotlib>=3.7.0",