def edit_state_action_to_softmax( self, state: S, action_value_dict: Mapping[A, float] ) -> None: self.policy_data[state] = get_softmax_action_probs( action_value_dict )
def sp_func(s: S, qf=qf, state_action_func=state_action_func, softmax=softmax, epsilon=epsilon) -> Mapping[A, float]: av_dict = {a: qf((s, a)) for a in state_action_func(s)} return get_softmax_action_probs(av_dict) if softmax else\ get_epsilon_action_probs(av_dict, epsilon)
def get_soft_policy_from_qf_dict(qf_dict: SAf, softmax: bool, epsilon: float) -> Policy: if softmax: ret = Policy( {s: get_softmax_action_probs(v) for s, v in qf_dict.items()}) else: ret = Policy({ s: get_epsilon_action_probs(v, epsilon) for s, v in qf_dict.items() }) return ret
def get_softmax_policy_from_qf( qf_dict: Mapping[S, Mapping[A, float]]) -> Policy: return Policy({s: get_softmax_action_probs(v) for s, v in qf_dict.items()})
def sp_func(s: S, softmax=softmax, epsilon=epsilon) -> Mapping[A, float]: av_dict = get_act_value_dict_from_state(s) return get_softmax_action_probs(av_dict) if softmax else\ get_epsilon_action_probs(av_dict, epsilon)