problem.py

   1 #!/usr/bin/env python
   2
   3 # Any copyright is dedicated to the Public Domain.
   4 # https://creativecommons.org/publicdomain/zero/1.0/
   5
   6 # Written by Francois Fleuret <francois@fleuret.org>
   7
   8
   9 class Problem:
  10     def nb_token_values(self):
  11         pass
  12
  13     # returns two tensors nb x D and nb x D'
  14     def generate_prompts_and_answers(self, nb):
  15         pass
  16
  17     # save a file to vizualize quizzes, you can save a txt or png file
  18     def save_quizzes(
  19         self,
  20         result_dir,
  21         filename_prefix,
  22         prompts,
  23         answers,
  24         predicted_prompt=None,
  25         predicted_answers=None,
  26     ):
  27         pass