Fix/fallback to 1 when weights doesn't exist

open-spaced-repetition · Jan 9, 2025 · 60e4d39 · 60e4d39
1 parent 24dfb16
commit 60e4d39
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -1551,6 +1551,8 @@ def evaluate(self, save_to_file=True):
             lambda row: -np.log(row["p"]) if row["y"] == 1 else -np.log(1 - row["p"]),
             axis=1,
         )
+        if "weights" not in self.dataset.columns:
+            self.dataset["weights"] = 1
         self.dataset["log_loss"] = (
             self.dataset["log_loss"]
             * self.dataset["weights"]