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 function make_graph () {
  27     CLUSTER1=($(grep ^0 clusters.dat))
  28     CLUSTER2=($(grep ^1 clusters.dat))
  29     CLUSTER3=($(grep ^2 clusters.dat))
  30
  31     gnuplot <<EOF
  32 set terminal pngcairo truecolor size 1024,768
  33 set output "$1"
  34 set size ratio 1
  35 set key out vert
  36 set key left top
  37 set object 1 ellipse center ${CLUSTER1[1]}, ${CLUSTER1[2]} size ${CLUSTER1[3]}, ${CLUSTER1[4]} angle 0 front fs empty bo 0 lw 1
  38 set object 2 ellipse center ${CLUSTER2[1]}, ${CLUSTER2[2]} size ${CLUSTER2[3]}, ${CLUSTER2[4]} angle 0 front fs empty bo 0 lw 1
  39 set object 3 ellipse center ${CLUSTER3[1]}, ${CLUSTER3[2]} size ${CLUSTER3[3]}, ${CLUSTER3[4]} angle 0 front fs empty bo 0 lw 1
  40 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", \
  41                           "< grep ^1 associated_clusters.dat" using 2:3 w p lc rgb "#00c000" pt 6      ps 2.0 title "Cluster 2", \
  42                           "< grep ^2 associated_clusters.dat" using 2:3 w p lc rgb "#0000c0" pt 6      ps 2.0 title "Cluster 3", \
  43                           "< grep ^0 points.dat"              using 2:3 w p lc rgb "#e00000" pt 7      ps 1.0 title "Class 1", \
  44                           "< grep ^1 points.dat"              using 2:3 w p lc rgb "#00c000" pt 7      ps 1.0 title "Class 2", \
  45                           "< grep ^0 clusters.dat"            using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
  46                           "< grep ^0 clusters.dat"            using 2:3 w p lc rgb "#e00000" pt 2 lw 4 ps 4.0 title "Centroid 1", \
  47                           "< grep ^1 clusters.dat"            using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
  48                           "< grep ^1 clusters.dat"            using 2:3 w p lc rgb "#00c000" pt 2 lw 4 ps 4.0 title "Centroid 2", \
  49                           "< grep ^2 clusters.dat"            using 2:3 w p lc rgb "#ffffff" pt 2 lw 9 ps 4.0 notitle, \
  50                           "< grep ^2 clusters.dat"            using 2:3 w p lc rgb "#0000c0" pt 2 lw 4 ps 4.0 title "Centroid 3"
  51 EOF
  52 }
  53
  54 make -j -k
  55
  56 ./clueless-kmean standard
  57 make_graph result-standard.png
  58
  59 ./clueless-kmean clueless
  60 make_graph result-clueless.png