Explorar el Código

adjust parameters

Xi Ruoyao hace 4 años
padre
commit
3ce4c1d0ae
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  1. 2 2
      main.py

+ 2 - 2
main.py

@@ -84,10 +84,10 @@ if __name__ == '__main__':
             print("radius = %f m" % rad)
 
         yaw = yaws[class_member_mask]
-        db_yaw = DBSCAN(eps = 0.1, min_samples=100,
+        db_yaw = DBSCAN(eps = 0.05, min_samples=30,
                 metric = metric.ang_distance).fit(yaw.reshape(-1, 1))
         lbs = set(db_yaw.labels_)
-        arrow_colors = [plt.cm.Spectral(each)
+        arrow_colors = [plt.cm.Wistia(each)
                           for each in np.linspace(0, 1, len(lbs))]
         for l in lbs:
             if l != -1: