Adjusted Genetic Algor

Author: Guled

Example/MLKit/GameScene.swift

@@ -34,6 +34,10 @@ class GameScene: SKScene, SKPhysicsContactDelegate {
     /// The pipe that is in front of the bird
     var currentPipe: Int = 0
 
+    var maxFitness: Float = 0
+
+    var maxBird: FlappyGenome?
+
     var lastBestGen: [FlappyGenome] = []
 
     // END of ADDITIONS
@@ -213,7 +217,7 @@ class GameScene: SKScene, SKPhysicsContactDelegate {
         goalArea.name = "GOAL"
         goalArea.fillColor = SKColor.red
         goalArea.position = pipeUp.position
-        goalArea.position.y += 230
+        goalArea.position.y += 270
         // END of ADDITIONS
 
         pipePair.addChild(pipeUp)
@@ -295,8 +299,14 @@ class GameScene: SKScene, SKPhysicsContactDelegate {
             lastBestGen = (flappyBirdGenerationContainer?.filter({$0.fitness >= 4}))!
         }
 
+
+        if (currentBird?.fitness)! > maxFitness {
+            maxFitness = (currentBird?.fitness)!
+            maxBird = currentBird
+        }
+
         // If we have hit the 20th bird, we need to move on to the next generation
-        if currentFlappy == 10 {
+        if currentFlappy == 20 {
 
             print("GENERATING NEW GEN!")
 
@@ -306,8 +316,9 @@ class GameScene: SKScene, SKPhysicsContactDelegate {
             var newGen: [FlappyGenome] = []
 
             newGen += lastBestGen
+            newGen.append(maxBird!)
 
-            while newGen.count < 10 {
+            while newGen.count < 20 {
 
                 // Select the best parents
                 let parents = PopulationManager.selectParents(genomes: flappyBirdGenerationContainer!)
@@ -406,7 +417,7 @@ class GameScene: SKScene, SKPhysicsContactDelegate {
             let normalizedDistanceOfNextPipe = (distanceOfNextPipe - 3)/(725-3)
 
             // Bird Y position
-            let birdYPos = bird.position.y/CGFloat(800)
+            let birdYPos = bird.position.y/CGFloat(880)
 
             // Measure how close the bird is to the gap between the pipes
             let posToGap = pipes.children[0].children[2].position.y - bird.position.y
@@ -459,7 +470,7 @@ class GameScene: SKScene, SKPhysicsContactDelegate {
 
         if moving.speed > 0 {
 
-            if y >= 800 {
+            if y >= 880 {
                 moving.speed = 0
 
                 bird.physicsBody?.collisionBitMask = worldCategory

Example/MLKit/GameViewController.swift

@@ -37,12 +37,15 @@ extension SKNode {
 // Genome that represents a Flappy Bird
 public class FlappyGenome: Genome {
 
+    /// Genotype representation of the genome.
+    public var genotypeRepresentation: [Float]
+
+    public var fitness: Float = 0
+    
     public var brain: NeuralNet?
 
     public init(genotype: [Float], network: NeuralNet) {
 
-        super.init(genotype: genotype)
-
         self.genotypeRepresentation = genotype
         self.brain = network
     }
@@ -64,7 +67,7 @@ class GameViewController: UIViewController {
         // Create First Generation of Flappy Birds
         var generation1: [FlappyGenome] = []
 
-        for _ in 1...10 {
+        for _ in 1...20 {
 
             let brain = NeuralNet()
 

Example/Tests/GeneticSpec.swift

@@ -15,11 +15,27 @@ import Nimble
 
 class GeneticSpec: QuickSpec {
 
+
+    public struct FakeGenome: Genome {
+
+        var genotypeRepresentation: [Float]
+
+        var fitness: Float
+
+        public init(genotype: [Float]) {
+
+            self.genotypeRepresentation = genotype
+            self.fitness = 0
+        }
+
+    }
+
+
     override func spec() {
 
         it("Should be able to produce a unique genotype after swap mutation process.") {
 
-            let fakeGenome: Genome = Genome(genotype: [1.0, 2.0, 3.0])
+            var fakeGenome: FakeGenome = FakeGenome(genotype: [1.0, 2.0, 3.0])
 
             let oldGenotype = fakeGenome.genotypeRepresentation
 
@@ -32,7 +48,7 @@ class GeneticSpec: QuickSpec {
 
         it("Should be able to produce a unique genotype after insert mutation process. ") {
 
-            let fakeGenome: Genome = Genome(genotype: [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0])
+            var fakeGenome: FakeGenome = FakeGenome(genotype: [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0])
 
             let oldGenotype = fakeGenome.genotypeRepresentation
 
@@ -44,7 +60,7 @@ class GeneticSpec: QuickSpec {
 
         it("Should be able to produce a unique genotype after inverse mutation process. ") {
 
-            let fakeGenome: Genome = Genome(genotype: [1.0, 2.0, 3.0])
+            var fakeGenome: FakeGenome = FakeGenome(genotype: [1.0, 2.0, 3.0])
 
             let oldGenotype = fakeGenome.genotypeRepresentation
 
@@ -55,7 +71,7 @@ class GeneticSpec: QuickSpec {
 
         it("Should be able to produce a unique genotype after scramble mutation process. ") {
 
-            let fakeGenome: Genome = Genome(genotype: [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0])
+            var fakeGenome: FakeGenome = FakeGenome(genotype: [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0])
 
             let oldGenotype = fakeGenome.genotypeRepresentation
 

MLKit/Classes/Genetic Algorithms/Genome.swift

@@ -8,25 +8,14 @@
 
 import Foundation
 
-
 /// Blueprint for a Genome. It is encouraged to inherit from this base class in order to accomodate for your needs.
-open class Genome {
+public protocol Genome {
 
     /// Genotype representation of the genome.
-    public var genotypeRepresentation: [Float]
+    var genotypeRepresentation: [Float] { get set }
 
     /// Fitness of a particular genome.
-    public var fitness: Float = 0
-
-    public init(genotype: [Float]) {
-
-        self.genotypeRepresentation = genotype
-    }
+    var fitness: Float { get set }
 
-    // Method used to assign a fitness value for a Genome
-    /*  This method is not given in the framework as there is no way to tell how you will evaluate fitness within your own project. Please implement your own 'generateFitness' method. An example is given in the Example Flappy Bird Game Project.
-    open func generateFitness() {
-
-    }
-    */
 }
+

README.md

@@ -7,7 +7,7 @@ MLKit is a simple machine learning framework written in Swift. Currently MLKit f
 [![License](https://img.shields.io/cocoapods/l/MLKit.svg?style=flat)](https://cocoapods.org/pods/MachineLearningKit)
 [![Platform](https://img.shields.io/cocoapods/p/MLKit.svg?style=flat)](https://cocoapods.org/pods/MachineLearningKit)
 
-[MachineLearningKit Reference](http://cocoadocs.org/docsets/MachineLearningKit/0.1.4/index.html)
+[MachineLearningKit Reference](http://cocoadocs.org/docsets/MachineLearningKit/0.1.5/)
 
 ## Requirements