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