def extract_state(self, state):
obs = np.zeros((7, 4, 15), dtype=int)
encode_hand(obs[:3], state['hand'])
encode_target(obs[3], state['target'])
encode_hand(obs[4:], state['others_hand'])
legal_action_id = self.get_legal_actions()
extrated_state = {'obs': obs, 'legal_actions': legal_action_id}
return extrated_state
def _extract_state(self, state):
obs = np.zeros((4, 4, 15), dtype=int)
encode_hand(obs[:3], state['hand'])
encode_target(obs[3], state['target'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [
a for a in state['legal_actions']
]
extracted_state['action_record'] = self.action_recorder
return extracted_state
def _extract_state(self, state):
obs = np.zeros((7, 4, 15), dtype=int)
encode_hand(obs[:3], state['hand'])
encode_target(obs[3], state['target'])
encode_hand(obs[4:], state['others_hand'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
if self.allow_raw_data:
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [a for a in state['legal_actions']]
if self.record_action:
extracted_state['action_record'] = self.action_recorder
return extracted_state
def _extract_state(self, state):
obs = np.zeros((3, 4, 15), dtype=int)
targetEncoding = np.zeros([19])
encode_hand(obs[:3], state['hand'])
encode_target(targetEncoding, state['target'])
obs = obs[1:].flatten()
next_player_amount = [0, 0, 0, 0]
nextAmountOfCards = state['next_player_amount']
if nextAmountOfCards >= 4:
next_player_amount[3] = 1
else:
next_player_amount[nextAmountOfCards - 1] = 1
obs = np.concatenate((obs, next_player_amount, targetEncoding))
# encode_hand(obs[4:], state['others_hand'])
legal_action_id = self._get_legal_actions()
extracted_state = {'obs': obs, 'legal_actions': legal_action_id}
if self.allow_raw_data:
extracted_state['raw_obs'] = state
extracted_state['raw_legal_actions'] = [
a for a in state['legal_actions']
]
if self.record_action:
extracted_state['action_record'] = self.action_recorder
return extracted_state
def test_encode_target(self):
encoded_target = np.zeros((4, 15), dtype=int)
target = 'r-1'
encode_target(encoded_target, target)
self.assertEqual(encoded_target[0][1], 1)