code: def load(self, train=True, test=True, shuffle=True) -> tuple:
          
          return self.__load(self.__load_files, train, test, shuffle=shuffle)
docs: Load the vectorized representations of the stored data files
          Args:
              train: Whether to load train data
              test: Whether to load test data