Feat/expose gamma to optimizer.train() (#162)

open-spaced-repetition · Jan 24, 2025 · be0c7da · be0c7da
1 parent 3f9a8ee
commit be0c7da
Show file tree

Hide file tree

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

diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -1233,6 +1233,7 @@ def train(
         self,
         lr: float = 4e-2,
         n_epoch: int = 5,
+        gamma: float = 1.0,
         batch_size: int = 512,
         verbose: bool = True,
         split_by_time: bool = False,
@@ -1263,6 +1264,7 @@ def train(
                     self.init_w,
                     n_epoch=n_epoch,
                     lr=lr,
+                    gamma=gamma,
                     batch_size=batch_size,
                     float_delta_t=self.float_delta_t,
                     enable_short_term=self.enable_short_term,
@@ -1287,6 +1289,7 @@ def train(
                 self.init_w,
                 n_epoch=n_epoch,
                 lr=lr,
+                gamma=gamma,
                 batch_size=batch_size,
                 float_delta_t=self.float_delta_t,
                 enable_short_term=self.enable_short_term,