def __init__(self): super(Mapper, self).__init__() self.qlss, self.qrss, self.num_images = {}, {}, {} try: tree_ser = pickle.load(open(os.environ['TREE_SER_FN'])) except KeyError: self.dp = None else: self.dp = tree_features.make_predict([tree_ser]) self.min_sparsity = os.environ.get('MIN_SPARSITY', .05) self.max_root_buffer = os.environ.get('MAX_ROOT_BUFFER', 32) self.root_count = {}
def __init__(self): self.tp = tree_features.make_predict(pickle.load(open(os.environ["TREES_SER_FN"]))) self.integral_type = os.environ["INTEGRAL_TYPE"]