From 279cb7a703aac1278b6aecb8c0b23eb35d2f11a2 Mon Sep 17 00:00:00 2001 From: troyyyyy <49091847+troyyyyy@users.noreply.github.com> Date: Mon, 5 Dec 2022 16:36:08 +0800 Subject: [PATCH] Add zoopt option --- abducer/abducer_base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/abducer/abducer_base.py b/abducer/abducer_base.py index 0a7f6cb..d0baf1f 100644 --- a/abducer/abducer_base.py +++ b/abducer/abducer_base.py @@ -51,8 +51,7 @@ class AbducerBase(abc.ABC): cols = np.array(range(len(B[0]))) cols = np.expand_dims(cols, axis = 0).repeat(axis = 0, repeats = len(B)) return 1 - np.prod(A[rows, cols, B], axis = 1) - - + def get_cost_list(self, pred_res, pred_res_prob, candidates): if self.dist_func == 'hamming': return self.hamming_dist(pred_res, candidates)