Browse Source

Add zoopt option

pull/3/head
troyyyyy GitHub 3 years ago
parent
commit
279cb7a703
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      abducer/abducer_base.py

+ 1
- 2
abducer/abducer_base.py View File

@@ -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)


Loading…
Cancel
Save