Added figures

[culture.git] / quizz_machine.py

diff --git a/quizz_machine.py b/quizz_machine.py
index 4e7576e..697f27e 100755 (executable)
--- a/quizz_machine.py
+++ b/quizz_machine.py
@@ -139,6 +139,7 @@ class QuizzMachine:
         self.train_w_quizzes = self.problem.generate_token_sequences(
             nb_train_samples
         ).to(device)
         self.train_w_quizzes = self.problem.generate_token_sequences(
             nb_train_samples
         ).to(device)

+

         self.test_w_quizzes = self.problem.generate_token_sequences(nb_test_samples).to(
             device
         )
         self.test_w_quizzes = self.problem.generate_token_sequences(nb_test_samples).to(
             device
         )