コード例 #1
0
ファイル: data_loader.py プロジェクト: omossad/fat-albert
    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!')
コード例 #2
0
    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!'