Robust estimation

Author: danjjl

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "szcore-evaluation"
-version = "0.0.5"
+version = "0.0.6"
 description = "Compare szCORE compliant annotations of EEG datasets of people with epilelpsy."
 authors = [
     { name = "Jonathan Dan", email = "jonathan.dan@epfl.ch" }

src/szcore_evaluation/evaluate.py

@@ -97,8 +97,14 @@ def evaluate_dataset(
                 hyp = Annotation(np.zeros_like(ref.mask), ref.fs)
 
             # Compute evaluation
-            sample_score = scoring.SampleScoring(ref, hyp)
-            event_score = scoring.EventScoring(ref, hyp)
+            try:
+                sample_score = scoring.SampleScoring(ref, hyp)
+                event_score = scoring.EventScoring(ref, hyp)
+            except ValueError as e:
+                print(f"Error in {ref_tsv}: {e}")
+                hyp = Annotation(np.zeros_like(ref.mask), ref.fs)
+                sample_score = scoring.SampleScoring(ref, hyp)
+                event_score = scoring.EventScoring(ref, hyp)
 
             # Store results
             sample_results[subject.name] += Result(sample_score)