From: François Fleuret <francois@fleuret.org>
Date: Wed, 17 Jul 2024 03:53:27 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=b41d98a66bdeb0282a8bf2268d658ab60081a819;p=culture.git

Update.
---

diff --git a/quiz_machine.py b/quiz_machine.py
index 1168921..14e8739 100755
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -365,7 +365,7 @@ class QuizMachine:
                 seq_logproba=seq_logproba,
                 temperature=1.0,
                 deterministic_synthesis=deterministic_synthesis,
-                progress_bar_desc=None,
+                progress_bar_desc="accuracy",
                 device=self.device,
             )
 
@@ -402,7 +402,7 @@ class QuizMachine:
             model.test_w_quizzes[:2000], log_prefix="test"
         )
 
-        n_test_forward = model.test_w_quizzes[:, 0] == self.token_forward
+        n_test_forward = model.test_w_quizzes[:2000, 0] == self.token_forward
 
         forward_test_correct = test_correct[n_test_forward]