From f86b04c9769fe2b36d247be3fecac14f410d370f Mon Sep 17 00:00:00 2001
From: "valentin.emiya" <valentin.emiya@lif.univ-mrs.fr>
Date: Thu, 3 Dec 2020 09:16:25 +0100
Subject: [PATCH] add rank and mask_ratio features in results
---
python/tffpy/experiments/exp_solve_tff.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/python/tffpy/experiments/exp_solve_tff.py b/python/tffpy/experiments/exp_solve_tff.py
index 7032490..44ba2e1 100644
--- a/python/tffpy/experiments/exp_solve_tff.py
+++ b/python/tffpy/experiments/exp_solve_tff.py
@@ -953,7 +953,9 @@ def perf_measures(task_params, source_data, problem_data,
t_subreg=solved_data['t_subreg']
)
features = dict(mask_size=np.sum(gmtff.mask > 0),
- n_subregions=gmtff.n_areas)
+ mask_ratio=np.mean(gmtff.mask > 0),
+ n_subregions=gmtff.n_areas,
+ rank_sum=np.sum([s.size for s in gmtff.s_vec_list]))
return dict(**running_times, **sdr_res, **is_res, **features)
--
GitLab