OptionClass.py 文件源码

def __init__(self, init_predicate, term_predicate, policy, name="o", term_prob=0.01):
        '''
        Args:
            init_func (S --> {0,1})
            init_func (S --> {0,1})
            policy (S --> A)
        '''
        self.init_predicate = init_predicate
        self.term_predicate = term_predicate
        self.term_flag = False
        self.name = name
        self.term_prob = term_prob

        if type(policy) is defaultdict or type(policy) is dict:
            self.policy_dict = dict(policy)
            self.policy = self.policy_from_dict
        else:
            self.policy = policy