Skip to content

Commit 142e06f

Browse files
anna-grimanna-grim
andauthored
refactor: improved metric defaults (#146)
Co-authored-by: anna-grim <[email protected]>
1 parent 1153b6c commit 142e06f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/segmentation_skeleton_metrics/skeleton_metric.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,9 @@ def __init__(
146146
"ERL",
147147
"Normalized ERL",
148148
]
149-
self.metrics = pd.DataFrame(index=row_names, columns=col_names)
149+
self.metrics = pd.DataFrame(np.nan, index=row_names, columns=col_names)
150+
self.metrics["# Merges"] = 0
151+
self.metrics["# Splits"] = 0
150152
self.metrics["SWC Name"] = self.metrics.index
151153

152154
# --- Load Data ---
@@ -314,7 +316,6 @@ def run(self):
314316
util.update_txt(path, " # Splits: " + str(n_splits))
315317

316318
if self.fragment_graphs is not None:
317-
self.metrics["# Merges"].fillna(0, inplace=True)
318319
n_merges = self.metrics["# Merges"].sum()
319320
util.update_txt(path, " # Merges: " + str(n_merges))
320321

0 commit comments

Comments
 (0)