X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=test.sh;h=51735ee4a161dbfb90245d9b3f704fe8b62fe5b5;hb=60785fe128e964f3d124e9b49cce9154501e53cb;hp=b486b03720e166b26a5cfed3757c50b85a99052c;hpb=2455f83ba251602d5e04640067094f09f03aaa3d;p=clueless-kmeans.git

diff --git a/test.sh b/test.sh
index b486b03..51735ee 100755
--- a/test.sh
+++ b/test.sh
@@ -23,32 +23,38 @@
 
 set -e
 
-make -j -k
-
-./clueless-kmean
+function make_graph () {
+    CLUSTER1=($(grep ^0 clusters.dat))
+    CLUSTER2=($(grep ^1 clusters.dat))
+    CLUSTER3=($(grep ^2 clusters.dat))
 
-CLUSTER1=($(grep ^0 clusters.dat))
-CLUSTER2=($(grep ^1 clusters.dat))
-CLUSTER3=($(grep ^2 clusters.dat))
-
-gnuplot <<EOF
+    gnuplot <<EOF
 set terminal pngcairo truecolor size 1024,768
-set output "result.png"
+set output "$1"
 set size ratio 1
 set key out vert
 set key left top
 set object 1 ellipse center ${CLUSTER1[1]}, ${CLUSTER1[2]} size ${CLUSTER1[3]}, ${CLUSTER1[4]} angle 0 front fs empty bo 0 lw 1
 set object 2 ellipse center ${CLUSTER2[1]}, ${CLUSTER2[2]} size ${CLUSTER2[3]}, ${CLUSTER2[4]} angle 0 front fs empty bo 0 lw 1
 set object 3 ellipse center ${CLUSTER3[1]}, ${CLUSTER3[2]} size ${CLUSTER3[3]}, ${CLUSTER3[4]} angle 0 front fs empty bo 0 lw 1
-plot [-1.2:1.2][-1.2:1.2] "< grep ^0 associated_clusters.dat" using 2:3 w p lc rgb "#e00000" pt 6 ps 2.0 title "Cluster 1", \
-                          "< grep ^1 associated_clusters.dat" using 2:3 w p lc rgb "#00c000" pt 6 ps 2.0 title "Cluster 2", \
-                          "< grep ^2 associated_clusters.dat" using 2:3 w p lc rgb "#0000c0" pt 6 ps 2.0 title "Cluster 3", \
-                          "< grep ^0 points.dat"               using 2:3 w p lc rgb "#e00000" pt 7 ps 1.0 title "Class 1", \
-                          "< grep ^1 points.dat"               using 2:3 w p lc rgb "#00c000" pt 7 ps 1.0 title "Class 2", \
-                          "< grep ^0 clusters.dat"             using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
-                          "< grep ^0 clusters.dat"             using 2:3 w p lc rgb "#e00000" pt 2 lw 4 ps 4.0 title "Centroid 1", \
-                          "< grep ^1 clusters.dat"             using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
-                          "< grep ^1 clusters.dat"             using 2:3 w p lc rgb "#00c000" pt 2 lw 4 ps 4.0 title "Centroid 2", \
-                          "< grep ^2 clusters.dat"             using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
-                          "< grep ^2 clusters.dat"             using 2:3 w p lc rgb "#0000c0" pt 2 lw 4 ps 4.0 title "Centroid 3"
+plot [-1.2:1.2][-1.2:1.2] "< grep ^0 associated_clusters.dat" using 2:3 w p lc rgb "#e00000" pt 6      ps 2.0 title "Cluster 1", \
+                          "< grep ^1 associated_clusters.dat" using 2:3 w p lc rgb "#00c000" pt 6      ps 2.0 title "Cluster 2", \
+                          "< grep ^2 associated_clusters.dat" using 2:3 w p lc rgb "#0000c0" pt 6      ps 2.0 title "Cluster 3", \
+                          "< grep ^0 points.dat"              using 2:3 w p lc rgb "#e00000" pt 7      ps 1.0 title "Class 1", \
+                          "< grep ^1 points.dat"              using 2:3 w p lc rgb "#00c000" pt 7      ps 1.0 title "Class 2", \
+                          "< grep ^0 clusters.dat"            using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
+                          "< grep ^0 clusters.dat"            using 2:3 w p lc rgb "#e00000" pt 2 lw 4 ps 4.0 title "Centroid 1", \
+                          "< grep ^1 clusters.dat"            using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
+                          "< grep ^1 clusters.dat"            using 2:3 w p lc rgb "#00c000" pt 2 lw 4 ps 4.0 title "Centroid 2", \
+                          "< grep ^2 clusters.dat"            using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
+                          "< grep ^2 clusters.dat"            using 2:3 w p lc rgb "#0000c0" pt 2 lw 4 ps 4.0 title "Centroid 3"
 EOF
+}
+
+make -j -k
+
+./clueless-kmean standard
+make_graph result-standard.png
+
+./clueless-kmean clueless
+make_graph result-clueless.png