Deal with NaN values

danjjl · danjjl · commit c41640b86cc2 · 2024-11-13T13:48:02.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "szcore-evaluation"
-version = "0.0.3"
+version = "0.0.4"
 description = "Compare szCORE compliant annotations of EEG datasets of people with epilelpsy."
 authors = [
     { name = "Jonathan Dan", email = "jonathan.dan@epfl.ch" }
diff --git a/src/szcore_evaluation/evaluate.py b/src/szcore_evaluation/evaluate.py
@@ -116,10 +116,10 @@ def evaluate_dataset(
             (aggregated_sample_results, aggregated_event_results),
         ):
             for metric in ["sensitivity", "precision", "f1", "fpRate"]:
-                aggregated_result[metric] = np.mean(
+                aggregated_result[metric] = np.nanmean(
                     [getattr(x, metric) for x in result_builder.values()]
                 )
-                aggregated_result[f"{metric}_std"] = np.std(
+                aggregated_result[f"{metric}_std"] = np.nanstd(
                     [getattr(x, metric) for x in result_builder.values()]
                 )
     else:

Original file line number	Diff line number	Diff line change
`@@ -116,10 +116,10 @@ def evaluate_dataset(`
`116`	`116`	`(aggregated_sample_results, aggregated_event_results),`
`117`	`117`	`):`
`118`	`118`	`for metric in ["sensitivity", "precision", "f1", "fpRate"]:`
`119`		`- aggregated_result[metric] = np.mean(`
	`119`	`+ aggregated_result[metric] = np.nanmean(`
`120`	`120`	`[getattr(x, metric) for x in result_builder.values()]`
`121`	`121`	`)`
`122`		`- aggregated_result[f"{metric}_std"] = np.std(`
	`122`	`+ aggregated_result[f"{metric}_std"] = np.nanstd(`
`123`	`123`	`[getattr(x, metric) for x in result_builder.values()]`
`124`	`124`	`)`
`125`	`125`	`else:`