create "tensor" if it doesn't exist during evaluation

open-spaced-repetition · Jan 9, 2025 · 8ef0528 · 8ef0528
1 parent e70dbfb
commit 8ef0528
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/fsrs_optimizer/fsrs_optimizer.py b/src/fsrs_optimizer/fsrs_optimizer.py
@@ -1541,6 +1541,11 @@ def moving_average(data, window_size=365 // 20):
 
     def evaluate(self, save_to_file=True):
         my_collection = Collection(DEFAULT_PARAMETER, self.float_delta_t)
+        if "tensor" not in self.dataset.columns:
+            self.dataset["tensor"] = self.dataset.progress_apply(
+                lambda x: lineToTensor(list(zip([x["t_history"]], [x["r_history"]]))[0]),
+                axis=1,
+            )
         stabilities, difficulties = my_collection.batch_predict(self.dataset)
         self.dataset["stability"] = stabilities
         self.dataset["difficulty"] = difficulties