Browse Source

output sub-clusters size

Xi Ruoyao 4 years ago
parent
commit
694f386e88
1 changed files with 3 additions and 2 deletions
  1. 3 2
      main.py

+ 3 - 2
main.py

@@ -86,8 +86,9 @@ if __name__ == '__main__':
                           for each in np.linspace(0, 1, len(lbs))]
         for l in lbs:
             if l != -1:
-                print("  sub-cluster %d: avg yaw = %f" %
-                        (l, 180 / pi * np.average(yaw[(db_yaw.labels_ == l)])))
+                print("  sub-cluster %d: size = %d, avg yaw = %f" %
+                        (l, np.sum(db_yaw.labels_ == l),
+                         180 / pi * np.average(yaw[(db_yaw.labels_ == l)])))
 
         kwargs = {'width': 1e-5}
         for i in range(len(xy)):