evopop 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 43f55266a4ef8f115db03cc3dd2f075ce4189110
-  data.tar.gz: e66bcb2525067d93ae87c8c3355c5a1f32030d05
+  metadata.gz: 0da75754d1ffbe965685e6298e01c2e121d7f2d9
+  data.tar.gz: 518b5ad7c766dd68d25942b28aa94dab2df0958f
 SHA512:
-  metadata.gz: 5e9fda825a2389dde2ef427cbd2d8d283e8d03ca42227d8266a99244d01799d782a2ca9cd51615cbcf4687e01796c891125951e8654c549c40b5485c7be5d391
-  data.tar.gz: 18c4d946acf6c631db4784d10177ec01762f8a16524bbe794b1cdf97fb62002680c251fd939fa9c3a072182729555fb163055527c11ffd4e8a10d498aaf15880
+  metadata.gz: e70e9007decd9653104efc8f6e28f2d8f7a4495588e2def2a1fa31537cb54b31cbf588758ccfd9ddc58556819d0f3b69e20c3cd2de436329785212004b695183
+  data.tar.gz: 8757fab931278ad147efa01cb8c1440431c9c3af430f0c9b13e7a2005ee9bdf61215eb6ad3d79c93c479e2b3702224cb5799f8d77c81146c155592251f81547f

data/lib/evopop/candidate.rb

@@ -5,7 +5,7 @@ class Candidate
   attr_accessor :dna, :fitness
 
   # Simple initialization of candidate object.
-  def initialize
-    @dna = []
+  def initialize(dna=[])
+    @dna = dna
   end
-end
+end

data/lib/evopop/crossover.rb

@@ -0,0 +1,45 @@
+# Represents a collection of well known crossover functions.
+# 
+module Crossover
+
+  # Perform 1 point crossover for a pair of candidates at the ordinal.
+  # http://en.wikipedia.org/wiki/Crossover_(genetic_algorithm)#One-point_crossover
+  def self.one_point(candidates, params)
+    ordinal = params[:ordinal]
+
+    # Compose the dna of the first child from the first chunk of the 
+    # first candidate and the second chunk of the second candidate
+    dna0_left = candidates[0].dna.take(ordinal)
+    dna1_right = candidates[1].dna.drop(ordinal)
+    
+    # Compose the dna of the second child from the first chunk of the 
+    # first candidate and the second chunk of the second candidate
+    dna1_left = candidates[1].dna.take(ordinal)
+    dna0_right = candidates[0].dna.drop(ordinal)
+
+    # Initialize and assign DNA to children.
+    children = [Candidate.new(dna = dna0_left + dna1_right), 
+                Candidate.new(dna = dna1_left + dna0_right)]
+
+    return children
+  end
+
+  # Perform n_point crossover for a pair of candidates. Will output two children from the n_point crossover.
+  # 
+  # Example:
+  # n_point
+  def self.n_point(candidates, params)
+
+  end
+
+  def self.average(candidates, params)
+    child = Candidate.new
+    dna_length = candidates[0].dna.length
+    (0...dna_length).each { |j|
+      child.dna << (candidates[0].dna[j] + candidates[1].dna[j])/2.0 # Initialize the dna of the child with the average of the parents' dna.
+    }
+
+    return [child]
+  end
+
+end

data/lib/evopop/population.rb

@@ -10,7 +10,7 @@
 #   population.crossover
 #   population.mutate
 class Population
-  attr_accessor :candidates, :population_size, :max_generations, :initial_range_min, :initial_range_max, :mutation_range_min, :mutation_range_max, :mutation_num, :fitness_function, :dna_len, :average_fitness
+  attr_accessor :candidates, :population_size, :max_generations, :crossover_function, :crossover_params, :initial_range_min, :initial_range_max, :mutation_range_min, :mutation_range_max, :mutation_num, :fitness_function, :dna_len, :average_fitness
 
   # Initializes the attributes with default values. This is not guaranteed
   # to reach maxima.
@@ -24,7 +24,9 @@ class Population
     @mutation_range_max = 10
     @mutation_num = (0.10*@population_size).to_i
     @dna_len = 1
+    @crossover_params = {:ordinal => (@dna_len/2)}
 
+    @crossover_function = Crossover.method(:one_point)
     @fitness_function = Proc.new { |dna|
       Math.sin(dna[0])
     }
@@ -61,22 +63,31 @@ class Population
   end
 
   # Performs simple mechanism of crossover - in this case picks two
-  # random candidates in from a top percentile of the population and averages
-  # out their DNA per dimension, producing new offspring equal to the
+  # random candidates in from a top percentile of the population and 
+  # performs one point crossover, producing new offspring equal to the
   # population size attribute.
   def crossover
+    # Define the candidates that can have children.
     @candidates = @candidates.take((@population_size*0.75).to_i)
     
     new_generation = Array.new
+    
     (0...@population_size).each {|i|
+      # For each of the top 75% of the population take 2
       couple = @candidates.sample(2)
-      child = Candidate.new
-      (0...@dna_len).each {|j|
-        child.dna << (couple[0].dna[j] + couple[1].dna[j])/2.0 # Initialize the dna of the child with the average of the parents' dna.
-      } 
-      new_generation << child 
-    }
+      params = @crossover_params
 
+      children = @crossover_function.call(couple, params)
+
+      new_generation = new_generation + children
+      
+      # When we go above set population_size, take the first population_size
+      # candidates, ignore the rest.
+      if new_generation.length >= self.population_size
+        new_generation = new_generation.take(self.population_size)
+        break
+      end
+    }
     @candidates = new_generation
   end
 

data/lib/evopop.rb

@@ -1,5 +1,6 @@
 require 'evopop/population'
 require 'evopop/candidate'
+require 'evopop/crossover'
 
 class Evopop
 

data/test/test_population.rb

@@ -14,6 +14,8 @@ class PopulationTest < Test::Unit::TestCase
     population.mutation_range_min = -100.0
     population.mutation_range_max = 100.0
     population.mutation_num = 10
+    population.crossover_params = {:ordinal => (population.dna_len/2)}
+    population.crossover_function = Crossover.method(:one_point)
     population.fitness_function = Proc.new { |dna|
       Math.sin(dna[0]) + Math.cos(dna[1])
     }
@@ -38,6 +40,10 @@ class PopulationTest < Test::Unit::TestCase
     }
   end
 
+  # Simple test of the training function. Ensure that when training
+  # finishes the fitness of the ith element of the population is 
+  # less than or equal to the i-1th element of the population. I.e.
+  # fitness is becoming greater over the iteration of the popoulation.
   def test_train
     # Arrange: Initialize the population
     population = initialize_population
@@ -55,6 +61,8 @@ class PopulationTest < Test::Unit::TestCase
     }
   end
 
+  # Simple test to ensure that only the exact number of candidates in the
+  # population are mutated.
   def test_mutation
     # Arrange: Initialize the population
     population = initialize_population
@@ -76,10 +84,37 @@ class PopulationTest < Test::Unit::TestCase
     assert_equal(population.mutation_num, counter)
   end
 
-  def test_cross_over
+  # Simple
+  def test_one_point_crossover
     # Arrange: Initialize the population
     population = initialize_population
+    
+    # Act: Train and corssover the population a number of times
+    5.times {
+      population.train
+      population.crossover
+    }
 
+    # Assert: The initial average fitness is less than what occurs after 100 generations.
+    # This is to ensure that over generations the average fitness does indeed go up, given
+    # no mutation.
+    assert_equal(true, population.average_fitness[0] < population.average_fitness[population.average_fitness.length-1])
   end
   
+  def test_average_crossover
+    # Arrange: Initialize the population
+    population = initialize_population
+    population.crossover_function = Crossover.method(:average)
+
+    # Act: Train and corssover the population a number of times
+    100.times {
+      population.train
+      population.crossover
+    }
+
+    # Assert: The initial average fitness is less than what occurs after 100 generations.
+    # This is to ensure that over generations the average fitness does indeed go up, given
+    # no mutation.
+    assert_equal(true, population.average_fitness[0] < population.average_fitness[population.average_fitness.length-1])
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: evopop
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Elvin Lucero
@@ -19,6 +19,7 @@ files:
 - lib/evopop.rb
 - lib/evopop/population.rb
 - lib/evopop/candidate.rb
+- lib/evopop/crossover.rb
 - test/test_population.rb
 homepage: https://github.com/elvinlucero/evopop
 licenses: