def __init__(self): self.load_me_stories = story_loader.StoryLoader() self.movies_map = dict() self.qa_list = list() self.data_split = dict() self._populate_movie() self._populate_splits() self._populate_qa() print('Initialized MovieQA data loader!')
def __init__(self): self.load_me_stories = story_loader.StoryLoader() # Dictionary of all the movies self.movies_map = dict() # List of all the questions self.qa_list = list() # Dictionary for the splits (train, val, test) self.data_split = dict() # Create a map of (imdb_key, MovieInfo) and its inversed map. self._populate_movie() # To create a split of train and test set self._populate_splits() # To create a list of the question ansd answers self._populate_qa() print 'Initialized MovieQA data loader!'