update: simplify unnecessary 'np.diag(...)' in ft_t1

ealcobaca · Jan 7, 2024 · 0afd1fc · 0afd1fc
1 parent 1867b69
commit 0afd1fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymfe/complexity.py b/pymfe/complexity.py
@@ -2234,10 +2234,10 @@ def _recurse_radius_calc(ind_inst: int) -> float:
         ref_dists = (
             radiuses
             if agglomeration_reference == "center"
-            else (radiuses - radiuses[:, np.newaxis] + np.diag(radiuses))
+            else (radiuses - radiuses[:, np.newaxis])
         )
 
-        within_hyperspheres = orig_dist_mat < ref_dists
+        within_hyperspheres = orig_dist_mat <= ref_dists
         agg_priority_metric = np.sum(within_hyperspheres, axis=0) if agglomeration_metric == "mass" else radiuses
         sorted_sphere_inds = np.argsort(-agg_priority_metric)
         sphere_inst_count = np.ones(radiuses.size, dtype=int)