neataptic/docs/js/articles/classifycolors/neural.js

Architecture-free neural network library with genetic algorithm implementations

var Neat    = neataptic.Neat;
var Methods = neataptic.Methods;
var Network = neataptic.Network;

var set;
var neat;
var running = false;
var iteration = 0;

var PER_COLOR = 50;
// Possible colors: red, orange, yellow, green, blue, purple, pink and monochrome
var COLORS = ['red', 'green', 'blue'];
var network;

$( document ).ready(function(){
  set = createSet();
  visualiseSet();
});

function createNeat(){
  network = new Network(3, COLORS.length);
  /*neat = new Neat(3, COLORS.length, fitness, {
    mutation: [
      Methods.Mutation.ADD_NODE,
      Methods.Mutation.ADD_CONN,
      Methods.Mutation.MOD_WEIGHT,
      Methods.Mutation.MOD_BIAS,
      Methods.Mutation.SUB_NODE,
      Methods.Mutation.MOD_ACTIVATION
    ],
    mutationRate: 0.6,
    elitism: 5,
    popsize: 100,
  });*/
}

function visualiseSet(){
  $('.set').empty();
  for(color in COLORS){
    $('.set').append('<div class="row ' + COLORS[color] + '"><h3>' + COLORS[color] + '</h3></div>');
  }

  for(var item in set){
    item = set[item];
    $('.'+ item.color).append('<div id="circle" style="background-color:' + item.rgb + '"></div>');
  }
}

function visualiseGenomeSet(genome){
  $('.fittestset').empty();
  for(color in COLORS){
    $('.fittestset').append('<div class="row fittest' + COLORS[color] + '"><h3>' + COLORS[color] + '</h3></div>');
  }

  for(var item in set){
    item = set[item];
    var output = genome.activate(item.input);
    var max = Math.max.apply(null, output);
    var color = COLORS[output.indexOf(max)];

    $('.fittest'+ color).append('<div id="circle" style="background-color:' + item.rgb + '"></div>');
  }
}

function loop(){
  network.evolve(set, {
    iterations: 1,
    mutationRate: 0.6,
    elisitm: 5,
    popSize: 100,
    mutation: Methods.Mutation.FFW,
    cost: Methods.Cost.MSE
  });

  visualiseGenomeSet(network);

  $('.iteration').text(iteration);
  $('.bestfitness').text(network.test(set).error);

  iteration++;
  if(running) setTimeout(loop, 1);
}

// Thanks to https://github.com/davidmerfield/randomColor !!
function createSet(){
  var set = [];

  for(index in COLORS){
    var color = COLORS[index];

    var randomColors = randomColor({ hue : color, count: PER_COLOR, format: 'rgb'});

    for(var random in randomColors){
      var rgb = randomColors[random];
      random = rgb.substring(4, rgb.length-1).replace(/ /g, '').split(',');
      for(var y in random) random[y] = random[y]/255;

      var output = Array.apply(null, Array(COLORS.length)).map(Number.prototype.valueOf, 0);
      output[index] = 1;

      set.push({ input: random, output: output, color: color, rgb: rgb});
    }
  }

  return set;
}