X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=c5acea74bf24d63a393154162bf0fecc2850f747;hb=8e37a868ac7dfc1cb5e924790929c6eebabbeb94;hp=cb28a7d5d9addc2f85eb46930a74bd88763178c5;hpb=f78cdbad69a877df92df41094a9f3f1036a1582a;p=culture.git

diff --git a/main.py b/main.py
index cb28a7d..c5acea7 100755
--- a/main.py
+++ b/main.py
@@ -394,7 +394,10 @@ def create_c_quizzes(
             f"keep c_quizzes {to_keep.size(0)}/{new_c_quizzes.size(0)} ({to_keep.size(0)*100/new_c_quizzes.size(0):.02f}%) total {sum([ x.size(0) for x in kept])}/{nb_to_generate}"
         )
 
-    new_c_quizzes = torch.cat(kept, dim=0)[: nb_for_train + nb_for_test]
+    new_c_quizzes = torch.cat(kept, dim=0)
+    new_c_quizzes = new_c_quizzes[
+        torch.randperm(new_c_quizzes.size(0))[: nb_for_train + nb_for_test]
+    ]
 
     quizz_machine.store_c_quizzes(new_c_quizzes[:nb_for_train], for_train=True)
     quizz_machine.store_c_quizzes(new_c_quizzes[nb_for_train:], for_train=False)