114 lines
2.7 KiB
JavaScript
114 lines
2.7 KiB
JavaScript
|
/** Rename vars */
|
||
|
|
var Neat = neataptic.Neat;
|
||
|
|
var Methods = neataptic.Methods;
|
||
|
|
var Config = neataptic.Config;
|
||
|
|
var Architect = neataptic.Architect;
|
||
|
|
|
||
|
|
/** Turn off warnings */
|
||
|
|
Config.warnings = false;
|
||
|
|
|
||
|
|
/** Settings */
|
||
|
|
var WIDTH = $('#field').width();
|
||
|
|
var HEIGHT = 500;
|
||
|
|
|
||
|
|
var MAX_AREA = 10000;
|
||
|
|
var MIN_AREA = 400;
|
||
|
|
|
||
|
|
var RELATIVE_SIZE = 1.1;
|
||
|
|
var DECREASE_SIZE = 0.998;
|
||
|
|
|
||
|
|
var DETECTION_RADIUS = 150;
|
||
|
|
var FOOD_DETECTION = 3;
|
||
|
|
var PLAYER_DETECTION = 3;
|
||
|
|
|
||
|
|
var MIN_SPEED = 0.6;
|
||
|
|
var SPEED = 2;
|
||
|
|
|
||
|
|
var FOOD_AREA = 80;
|
||
|
|
var FOOD_AMOUNT = Math.round(WIDTH * HEIGHT * 4e-4);
|
||
|
|
|
||
|
|
// GA settings
|
||
|
|
var PLAYER_AMOUNT = Math.round(WIDTH * HEIGHT * 8e-5);
|
||
|
|
var ITERATIONS = 10e10;
|
||
|
|
var START_HIDDEN_SIZE = 0;
|
||
|
|
|
||
|
|
// Trained population
|
||
|
|
var USE_TRAINED_POP = true;
|
||
|
|
|
||
|
|
// Global vars
|
||
|
|
var neat;
|
||
|
|
|
||
|
|
/** Construct the genetic algorithm */
|
||
|
|
function initNeat(){
|
||
|
|
neat = new Neat(
|
||
|
|
1 + PLAYER_DETECTION * 3 + FOOD_DETECTION * 2,
|
||
|
|
2,
|
||
|
|
null,
|
||
|
|
{
|
||
|
|
mutation: [
|
||
|
|
Methods.Mutation.ADD_NODE,
|
||
|
|
Methods.Mutation.SUB_NODE,
|
||
|
|
Methods.Mutation.ADD_CONN,
|
||
|
|
Methods.Mutation.SUB_CONN,
|
||
|
|
Methods.Mutation.MOD_WEIGHT,
|
||
|
|
Methods.Mutation.MOD_BIAS,
|
||
|
|
Methods.Mutation.MOD_ACTIVATION,
|
||
|
|
Methods.Mutation.ADD_GATE,
|
||
|
|
Methods.Mutation.SUB_GATE,
|
||
|
|
Methods.Mutation.ADD_SELF_CONN,
|
||
|
|
Methods.Mutation.SUB_SELF_CONN,
|
||
|
|
Methods.Mutation.ADD_BACK_CONN,
|
||
|
|
Methods.Mutation.SUB_BACK_CONN
|
||
|
|
],
|
||
|
|
popsize: PLAYER_AMOUNT,
|
||
|
|
mutationRate: 0.3,
|
||
|
|
elitism: Math.round(0.1 * PLAYER_AMOUNT),
|
||
|
|
network: new Architect.Random(
|
||
|
|
1 + PLAYER_DETECTION * 3 + FOOD_DETECTION * 2,
|
||
|
|
START_HIDDEN_SIZE,
|
||
|
|
2
|
||
|
|
)
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
if(USE_TRAINED_POP){
|
||
|
|
neat.population = population;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Start the evaluation of the current generation */
|
||
|
|
function startEvaluation(){
|
||
|
|
players = [];
|
||
|
|
highestScore = 0;
|
||
|
|
|
||
|
|
for(var genome in neat.population){
|
||
|
|
genome = neat.population[genome];
|
||
|
|
new Player(genome);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/** End the evaluation of the current generation */
|
||
|
|
function endEvaluation(){
|
||
|
|
console.log('Generation:', neat.generation, '- average score:', neat.getAverage());
|
||
|
|
|
||
|
|
neat.sort();
|
||
|
|
var newPopulation = [];
|
||
|
|
|
||
|
|
// Elitism
|
||
|
|
for(var i = 0; i < neat.elitism; i++){
|
||
|
|
newPopulation.push(neat.population[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Breed the next individuals
|
||
|
|
for(var i = 0; i < neat.popsize - neat.elitism; i++){
|
||
|
|
newPopulation.push(neat.getOffspring());
|
||
|
|
}
|
||
|
|
|
||
|
|
// Replace the old population with the new population
|
||
|
|
neat.population = newPopulation;
|
||
|
|
neat.mutate();
|
||
|
|
|
||
|
|
neat.generation++;
|
||
|
|
startEvaluation();
|
||
|
|
}
|