squadenv.py 文件源码

def __init__(self, env_data_filename, mode="single", max_utterance_len=20, max_game_turns=20):
        with open(env_data_filename, mode="rb") as in_file:
            self._env_data = pickle.load(in_file)
        self._mode = mode
        self._max_game_turns = max_game_turns
        self._num_tokens = len(self._env_data["id2token"])
        self._num_entities = len(self._env_data["env_data"])
        self._entities = list(self._env_data["env_data"].keys())
        self.action_space = spaces.Discrete(self._num_tokens)
        self.observation_space = spaces.MultiDiscrete([[0, self._num_tokens] * max_utterance_len])
        self._last_entity = 0
        self._last_question = 0
        self._last_sequence = []
        self._game_turns = None
        self._game_score = 0