Update.

[culture.git] / problem.py

diff --git a/problem.py b/problem.py
index 8d973eb..0bc83a1 100755 (executable)
--- a/problem.py
+++ b/problem.py
@@ -7,15 +7,21 @@
 
 
 class Problem:
 
 
 class Problem:

-    # returns a nb x (L+1+L) long tensor where L is the length of one
-    # of the two states of a quizz
-    def generate_seq(self, nb):
+    def nb_token_values(self):

         pass
 
         pass
 

-    # save a file to vizualize quizzes, you can save a txt or png file
-    def save_quizzes(self, input, result_dir, filename_prefix, logger):
+    # returns two tensors nb x D and nb x D'
+    def generate_prompts_and_answers(self, nb):

         pass
 
         pass
 

-    # returns a pair (forward_tokens, backward_token)
-    def direction_tokens(self):
+    # save a file to vizualize quizzes, you can save a txt or png file
+    def save_quizzes(
+        self,
+        result_dir,
+        filename_prefix,
+        prompts,
+        answers,
+        predicted_prompt=None,
+        predicted_answers=None,
+    ):

         pass
         pass