36 lines
1011 B
Python
36 lines
1011 B
Python
''' 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
|