KohonenSet klist = new KohonenSet();
float colorList[][];
int numColors = 200;

void setup(){
  noStroke();
  size(660,500);
  createColorList();
  klist.kset = colorList;
  klist.initWeights(50, 50, 3);
  ellipseMode(CENTER);
}

void draw(){
  background(230);
  if(klist.mapMode == 0) klist.map();
  if(klist.mapMode == 1) klist.errorMap();
  klist.legend();
  klist.bmuLabels();
  klist.train();
}

void createColorList(){
  colorList = new float[numColors][3];
  for (int i=0; i<numColors; i++){
    for (int j=0; j<3; j++){
      colorList[i][j] = random(255);
    }
  }
}

void keyReleased(){
  if (key == '=') klist.mapMode = 0;
  if (key == '-') klist.mapMode = 1;
}