Feat: added incorrect answer generation
This commit is contained in:
parent
90d7df45af
commit
32eaece47b
35
incorrect_answer_generation.py
Normal file
35
incorrect_answer_generation.py
Normal file
@ -0,0 +1,35 @@
|
||||
''' This module contains the class
|
||||
for generating incorrect alternative
|
||||
answers for a given answer
|
||||
'''
|
||||
import gensim.downloader as api
|
||||
from nltk.tokenize import sent_tokenize, word_tokenize
|
||||
import random
|
||||
|
||||
|
||||
class IncorrectAnswerGenerator:
|
||||
''' This class contains the methods
|
||||
for generating the incorrect answers
|
||||
given an answer
|
||||
'''
|
||||
|
||||
def __init__(self):
|
||||
# model required to fetch similar words
|
||||
self.model = api.load("glove-wiki-gigaword-100")
|
||||
|
||||
def get_all_options_dict(self, answer, num_options):
|
||||
''' This method returns a dict
|
||||
of 'num_options' options out of
|
||||
which one is correct and is the answer
|
||||
'''
|
||||
similar_words = self.model.similar_by_word(answer, topn=15)[::-1]
|
||||
options_dict = dict()
|
||||
|
||||
for i in range(1, num_options + 1):
|
||||
options_dict[i] = similar_words[i - 1][0]
|
||||
|
||||
replacement_idx = random.randint(1, num_options)
|
||||
|
||||
options_dict[replacement_idx] = answer
|
||||
|
||||
return options_dict
|
||||
Loading…
x
Reference in New Issue
Block a user