b486b03720e166b26a5cfed3757c50b85a99052c
[clueless-kmeans.git] / test.sh
1 #!/bin/bash
2
3
4 # clueless-kmean is a variant of k-mean which enforces balanced
5 # distribution of classes in every cluster
6 #
7 # Copyright (c) 2013 Idiap Research Institute, http://www.idiap.ch/
8 # Written by Francois Fleuret <francois.fleuret@idiap.ch>
9 #
10 # This file is part of clueless-kmean.
11 #
12 # clueless-kmean is free software: you can redistribute it and/or
13 # modify it under the terms of the GNU General Public License
14 # version 3 as published by the Free Software Foundation.
15 #
16 # clueless-kmean is distributed in the hope that it will be useful,
17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19 # General Public License for more details.
20 #
21 # You should have received a copy of the GNU General Public License
22 # along with selector.  If not, see <http://www.gnu.org/licenses/>.
23
24 set -e
25
26 make -j -k
27
28 ./clueless-kmean
29
30 CLUSTER1=($(grep ^0 clusters.dat))
31 CLUSTER2=($(grep ^1 clusters.dat))
32 CLUSTER3=($(grep ^2 clusters.dat))
33
34 gnuplot <<EOF
35 set terminal pngcairo truecolor size 1024,768
36 set output "result.png"
37 set size ratio 1
38 set key out vert
39 set key left top
40 set object 1 ellipse center ${CLUSTER1[1]}, ${CLUSTER1[2]} size ${CLUSTER1[3]}, ${CLUSTER1[4]} angle 0 front fs empty bo 0 lw 1
41 set object 2 ellipse center ${CLUSTER2[1]}, ${CLUSTER2[2]} size ${CLUSTER2[3]}, ${CLUSTER2[4]} angle 0 front fs empty bo 0 lw 1
42 set object 3 ellipse center ${CLUSTER3[1]}, ${CLUSTER3[2]} size ${CLUSTER3[3]}, ${CLUSTER3[4]} angle 0 front fs empty bo 0 lw 1
43 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", \
44                           "< grep ^1 associated_clusters.dat" using 2:3 w p lc rgb "#00c000" pt 6 ps 2.0 title "Cluster 2", \
45                           "< grep ^2 associated_clusters.dat" using 2:3 w p lc rgb "#0000c0" pt 6 ps 2.0 title "Cluster 3", \
46                           "< grep ^0 points.dat"               using 2:3 w p lc rgb "#e00000" pt 7 ps 1.0 title "Class 1", \
47                           "< grep ^1 points.dat"               using 2:3 w p lc rgb "#00c000" pt 7 ps 1.0 title "Class 2", \
48                           "< grep ^0 clusters.dat"             using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
49                           "< grep ^0 clusters.dat"             using 2:3 w p lc rgb "#e00000" pt 2 lw 4 ps 4.0 title "Centroid 1", \
50                           "< grep ^1 clusters.dat"             using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
51                           "< grep ^1 clusters.dat"             using 2:3 w p lc rgb "#00c000" pt 2 lw 4 ps 4.0 title "Centroid 2", \
52                           "< grep ^2 clusters.dat"             using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
53                           "< grep ^2 clusters.dat"             using 2:3 w p lc rgb "#0000c0" pt 2 lw 4 ps 4.0 title "Centroid 3"
54 EOF