X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=problem.py;h=0bc83a12148c5fec2af3ed7b96edfce3b86054d5;hb=a8e608a50b84583ad624cdf69d7b34699557235b;hp=25ffc494b9c9fe6e97cfe824a0e224a0af9abcce;hpb=e2c3b8046c3fddef8aacb74cf5f848d42044897e;p=culture.git

diff --git a/problem.py b/problem.py
index 25ffc49..0bc83a1 100755
--- a/problem.py
+++ b/problem.py
@@ -7,11 +7,21 @@
 
 
 class Problem:
-    def generate_seq(self, nb):
+    def nb_token_values(self):
         pass
 
-    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
 
-    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