tree_stealer.py 文件源码

def test_cont_feature(self, query, feature, epsilon, min_val=None,
                          max_val=None):
        """
        FInd splits on a continuous feature
        """
        if min_val is None:
            min_val = feature.min_val
        if max_val is None:
            max_val = feature.max_val

        query_max = make_query(query, feature.name, max_val)
        max_id = self.predict(query_max)
        query_min = make_query(query, feature.name, min_val)
        min_id = self.predict(query_min)
        logging.log(DEBUG, '\tmin val {} got {}'.format(min_val, min_id))
        logging.log(DEBUG, '\tmax val {} got {}'.format(max_val, max_id))

        # search for any splitting thresholds
        thresholds = sorted(self.line_search(query, feature.name, min_val,
                                             max_val, min_id, max_id, epsilon))
        logging.log(DEBUG, '\tthresholds: {}'.format(thresholds))
        return thresholds