INNER CODE UNIT · Python
encode_qa_pairs
AMontgomerie/question_generator · questiongenerator.py:346
def encode_qa_pairs(self, questions: List[str], answers: List[str]) -> List[torch.tensor]:
"""Takes a list of questions and a list of answers and encodes them as a list of tensors."""
encoded_pairs = []
for question, answer in zip(questions, answers):
encoded_qa = self._encode_qa(question, answer)
encoded_pairs.append(encoded_qa.to(self.device))
return encoded_pairs
def get_scores(self, encoded_qa_pairs: List[torch.tensor]) -> List[float]:
"""Generates scores for a list of encoded QA pairs."""
scores = {}
for i in range(len(encoded_qa_pairs)):
scores[i] = self._evaluate_qa(encoded_qa_pairs[i])
return [