aboutsummaryrefslogtreecommitdiff
path: root/analysis/scripts/make_table.py
diff options
context:
space:
mode:
Diffstat (limited to 'analysis/scripts/make_table.py')
-rw-r--r--analysis/scripts/make_table.py15
1 files changed, 13 insertions, 2 deletions
diff --git a/analysis/scripts/make_table.py b/analysis/scripts/make_table.py
index c326775e..8070eecd 100644
--- a/analysis/scripts/make_table.py
+++ b/analysis/scripts/make_table.py
@@ -103,6 +103,7 @@ class Method(Enum):
rfpop = "rfpop"
segneigh = "segneigh"
wbs = "wbs"
+ zero = "zero"
# Methods that support multidimensional datasets
@@ -112,6 +113,7 @@ MULTIMETHODS = [
Method.ecp,
Method.kcpa,
Method.rbocpdms,
+ Method.zero,
]
# Multidimensional datasets
@@ -126,7 +128,13 @@ MULTIDATASETS = [
MISSING_DATASETS = [Dataset.uk_coal_employ]
# Methods that handle missing values
-MISSING_METHODS = [Method.bocpdms, Method.ecp, Method.kcpa, Method.prophet]
+MISSING_METHODS = [
+ Method.bocpdms,
+ Method.ecp,
+ Method.kcpa,
+ Method.prophet,
+ Method.zero,
+]
@dataclass
@@ -323,7 +331,10 @@ def average_results(results):
if any(r.score is None for r in dset_results):
to_remove.append(dataset)
if to_remove:
- warning("\nWarning: Filtering out datasets: %r due to incomplete results for some detectors.\n" % to_remove)
+ warning(
+ "\nWarning: Filtering out datasets: %r due to incomplete results for some detectors.\n"
+ % to_remove
+ )
results = list(filter(lambda r: not r.dataset in to_remove, results))
# check that we are now complete: for all datasets and all methods in the