This repository has been archived by the owner on Dec 11, 2018. It is now read-only.
/
genome_test.go
121 lines (103 loc) · 3.22 KB
/
genome_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package neat
import (
"fmt"
"log"
"math/rand"
"testing"
)
func GenomeUnitTest() {
fmt.Println("===== Genome Unit Test =====")
fmt.Println("\x1b[32m=Testing creating a new genome...\x1b[0m")
g0 := NewGenome(0, 3, 5, 0.0)
fmt.Println(g0.String())
fmt.Println("\x1b[32m=Testing mutation...\x1b[0m")
g0.MutatePerturb(1.0)
g0.MutateAddNode(1.0, ActivationSet["sigmoid"])
g0.MutateAddConn(1.0)
fmt.Println(g0.String())
fmt.Println("\x1b[32m=Testing crossover...\x1b[0m")
g1 := NewGenome(1, 3, 1, 0.0)
g1.MutatePerturb(1.0)
g1.MutateAddNode(1.0, ActivationSet["sigmoid"])
g1.MutateAddConn(1.0)
g1.MutatePerturb(1.0)
g1.MutateAddNode(1.0, ActivationSet["sigmoid"])
g1.MutateAddConn(1.0)
fmt.Println("Parent 1:")
fmt.Println(g1.String())
g2 := NewGenome(2, 3, 1, 0.0)
g2.MutatePerturb(1.0)
g2.MutateAddNode(1.0, ActivationSet["sigmoid"])
g2.MutateAddConn(1.0)
g2.MutatePerturb(1.0)
g2.MutateAddNode(1.0, ActivationSet["sigmoid"])
g2.MutateAddConn(1.0)
fmt.Println("Parent 2:")
fmt.Println(g2.String())
g3 := Crossover(3, g1, g2, 0.0)
fmt.Println("Child:")
fmt.Println(g3.String())
fmt.Println("\x1b[32m=Testing compatibility distance...\x1b[0m")
g4 := NewGenome(4, 3, 1, 0.0)
g5 := NewGenome(5, 3, 1, 0.0)
// before mutation (they should be fairly compatible)
fmt.Println(g4.String())
fmt.Println(g5.String())
fmt.Printf("Compatibility distance: %f\n", Compatibility(g4, g5, 1.0, 1.0))
// after 1 mutation (should be less compatible)
g4.MutatePerturb(1.0)
g4.MutateAddNode(1.0, ActivationSet["sigmoid"])
g4.MutateAddConn(1.0)
g4.MutatePerturb(1.0)
g4.MutateAddNode(1.0, ActivationSet["sigmoid"])
g4.MutateAddConn(1.0)
g5.MutatePerturb(1.0)
g5.MutateAddNode(1.0, ActivationSet["sigmoid"])
g5.MutateAddConn(1.0)
g5.MutatePerturb(1.0)
g5.MutateAddNode(1.0, ActivationSet["sigmoid"])
g5.MutateAddConn(1.0)
fmt.Println(g4.String())
fmt.Println(g5.String())
fmt.Printf("Compatibility distance: %f\n", Compatibility(g4, g5, 1.0, 1.0))
// after 2 mutation (should be less compatible)
g4.MutatePerturb(1.0)
g4.MutateAddNode(1.0, ActivationSet["sigmoid"])
g4.MutateAddConn(1.0)
g4.MutatePerturb(1.0)
g4.MutateAddNode(1.0, ActivationSet["sigmoid"])
g4.MutateAddConn(1.0)
g5.MutatePerturb(1.0)
g5.MutateAddNode(1.0, ActivationSet["sigmoid"])
g5.MutateAddConn(1.0)
g5.MutatePerturb(1.0)
g5.MutateAddNode(1.0, ActivationSet["sigmoid"])
g5.MutateAddConn(1.0)
fmt.Println(g4.String())
fmt.Println(g5.String())
fmt.Printf("Compatibility distance: %f\n", Compatibility(g4, g5, 1.0, 1.0))
// after 3 mutation (should be less compatible)
g4.MutatePerturb(1.0)
g4.MutateAddNode(1.0, ActivationSet["sigmoid"])
g4.MutateAddConn(1.0)
g4.MutatePerturb(1.0)
g4.MutateAddNode(1.0, ActivationSet["sigmoid"])
g4.MutateAddConn(1.0)
g5.MutatePerturb(1.0)
g5.MutateAddNode(1.0, ActivationSet["sigmoid"])
g5.MutateAddConn(1.0)
g5.MutatePerturb(1.0)
g5.MutateAddNode(1.0, ActivationSet["sigmoid"])
g5.MutateAddConn(1.0)
fmt.Println(g4.String())
fmt.Println(g5.String())
fmt.Printf("Compatibility distance: %f\n", Compatibility(g4, g5, 1.0, 1.0))
fmt.Println("\x1b[32m=Testing JSON export...\x1b[0m")
if err := g1.ExportJSON(true); err != nil {
log.Fatal(err)
}
}
func TestGenome(t *testing.T) {
rand.Seed(0)
GenomeUnitTest()
}