From: François Fleuret <francois@fleuret.org>
Date: Mon, 1 Jul 2024 08:42:36 +0000 (+0300)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=51540cefc448684d5086297d23e9a1805da4d405;p=culture.git

Update.
---

diff --git a/main.py b/main.py
index 11eb8fd..fd8ab41 100755
--- a/main.py
+++ b/main.py
@@ -79,6 +79,8 @@ parser.add_argument("--dropout", type=float, default=0.1)
 
 parser.add_argument("--deterministic_synthesis", action="store_true", default=False)
 
+parser.add_argument("--reverse_cleanup", action="store_true", default=False)
+
 parser.add_argument("--problem", type=str, default="sky")
 
 parser.add_argument("--nb_gpts", type=int, default=5)
@@ -418,6 +420,7 @@ def create_c_quizzes(
             nb_models_for_generation=args.nb_models_for_generation,
             models=models,
             mode=args.generation_mode,
+            reverse_cleanup=args.reverse_cleanup,
             min_ave_seq_logproba=min_ave_seq_logproba,
             n_epoch=n_epoch,
             result_dir=args.result_dir,
diff --git a/quizz_machine.py b/quizz_machine.py
index eae256b..806dde7 100755
--- a/quizz_machine.py
+++ b/quizz_machine.py
@@ -471,16 +471,18 @@ class QuizzMachine:
         models,
         mode,
         min_ave_seq_logproba,
+        reverse_cleanup,
         n_epoch,
         result_dir,
     ):
         model_for_generation = Gang(models, nb_models_for_generation, mode)
         models_for_validation = models
         return self.create_c_quizzes(
-            nb,
-            model_for_generation,
-            models_for_validation,
-            min_ave_seq_logproba,
-            n_epoch,
-            result_dir,
+            nb=nb,
+            model_for_generation=model_for_generation,
+            models_for_validation=models_for_validation,
+            min_ave_seq_logproba=min_ave_seq_logproba,
+            reverse_cleanup=reverse_cleanup,
+            n_epoch=n_epoch,
+            result_dir=result_dir,
         )