فهرست منبع

output sub-clusters size

Xi Ruoyao 6 سال پیش
والد
کامیت
694f386e88
1فایلهای تغییر یافته به همراه3 افزوده شده و 2 حذف شده
  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)):