Feat: Tied together question extraction and incorrect anwer generation
This commit is contained in:
parent
32eaece47b
commit
9574fa0932
31
question_generation_main.py
Normal file
31
question_generation_main.py
Normal file
@ -0,0 +1,31 @@
|
||||
'''This module ties together the
|
||||
questions generation and incorrect answer
|
||||
generation modules
|
||||
'''
|
||||
from question_extraction import QuestionExtractor
|
||||
from incorrect_answer_generation import IncorrectAnswerGenerator
|
||||
|
||||
class QuestionGeneration:
|
||||
'''This class contains the method
|
||||
to generate questions
|
||||
'''
|
||||
|
||||
def __init__(self, num_questions, num_options):
|
||||
self.num_questions = num_questions
|
||||
self.num_options = num_options
|
||||
self.question_extractor = QuestionExtractor(num_questions)
|
||||
|
||||
def generate_questions_dict(self, document):
|
||||
self.questions_dict = self.question_extractor.get_questions_dict(document)
|
||||
self.incorrect_answer_generator = IncorrectAnswerGenerator(document)
|
||||
|
||||
for i in range(1, self.num_questions+1):
|
||||
if i not in self.questions_dict:
|
||||
continue
|
||||
self.questions_dict[i]["options"] \
|
||||
= self.incorrect_answer_generator.get_all_options_dict(
|
||||
self.questions_dict[i]["answer"],
|
||||
self.num_options
|
||||
)
|
||||
|
||||
return self.questions_dict
|
||||
Loading…
x
Reference in New Issue
Block a user