Fix/init s out of order (#17)

.gitignore

@@ -148,4 +148,5 @@ meta
 *.tsv
 *.png
 *.json
-.fsrs_optimizer
+.fsrs_optimizer
+*.ipynb

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "FSRS-Optimizer"
-version = "4.7.0"
+version = "4.7.1"
 readme = "README.md"
 dependencies = [
     "matplotlib>=3.7.0",

src/fsrs_optimizer/fsrs_optimizer.py

@@ -554,12 +554,19 @@ def loss(stability):
                 plots.append(fig)
                 tqdm.write(str(rating_stability))
 
+        for a, b in ((1, 2), (2, 3), (3, 4)):
+            if a in rating_stability and b in rating_stability:
+                if rating_stability[a] > rating_stability[b]:
+                    if rating_count[a] > rating_count[b]:
+                        rating_stability[b] = rating_stability[a]
+                    else:
+                        rating_stability[a] = rating_stability[b]
+
         if len(rating_stability) == 0:
             raise Exception("Not enough data for pretraining!")
         elif len(rating_stability) == 1:
             init_stability = round(list(rating_stability.values())[0], 2)
-            for i in (0, 1, 2, 3):
-                self.init_w[i] = init_stability
+            self.init_w[0:4] = [init_stability] * 4
         elif len(rating_stability) == 4:
             for rating, stability in rating_stability.items():
                 self.init_w[rating-1] = round(stability, 2)