Fix Benchmark (#169)

tansey-lab · Dec 6, 2024 · a783d16 · a783d16
1 parent 602d1e4
commit a783d16
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/nuc2seg/postprocess.py b/src/nuc2seg/postprocess.py
@@ -549,6 +549,9 @@ def get_jaccard_truth_segment(row):
     method_to_truth_transcripts["truth_segment_id"] = method_to_truth_transcripts[
         "truth_segment_id"
     ].astype(int)
+    method_to_truth_transcripts["method_segment_id"] = method_to_truth_transcripts[
+        "method_segment_id"
+    ].astype(int)
 
     method_to_truth_transcripts_agg = method_to_truth_transcripts.groupby(
         "method_segment_id"
@@ -637,7 +640,11 @@ def get_jaccard_truth_segment(row):
         left_on="truth_segment_id",
         right_on="truth_segment_id",
         how="left",
-    ).merge(
+    )
+
+    results["method_segment_id"] = results["method_segment_id"].fillna(-1).astype(int)
+
+    results = results.merge(
         segmentation_errors[
             [
                 "n_confused_transcripts",