add dendrogram

Kapoorlabs-CAPED · Dec 30, 2023 · 981848f · 981848f
1 parent b9c30d0
commit 981848f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py
@@ -744,13 +744,17 @@ def create_analysis_tracklets(
     dividing_track_ids = subset_dividing["Track ID"].unique()
     dividing_count = len(dividing_track_ids)
     non_dividing_count = len(non_dividing_track_ids)
-    if non_dividing_count > dividing_count:
+    if non_dividing_count > dividing_count and class_ratio > 0:
         non_dividing_track_ids = random.sample(
             list(non_dividing_track_ids),
             int(min(non_dividing_count, dividing_count * class_ratio)),
         )
     else:
-        dividing_track_ids = random.sample(list(dividing_track_ids), non_dividing_count)
+        dividing_track_ids = random.sample(list(dividing_track_ids), dividing_count)
+        non_dividing_track_ids = random.sample(
+            list(non_dividing_track_ids), non_dividing_count
+        )
+
     print(
         f"Analysis data Dividing track counts {len(dividing_track_ids)}, Non Dividing track counts {len(non_dividing_track_ids)}"
     )

diff --git a/src/napatrackmater/_version.py b/src/napatrackmater/_version.py
@@ -1,2 +1,2 @@
-__version__ = version = "4.7.0"
-__version_tuple__ = version_tuple = (4, 7, 0)
+__version__ = version = "4.7.1"
+__version_tuple__ = version_tuple = (4, 7, 1)