Clojush-Tools/program_mean_gen_sizes.py at master · GraceWoolson/Clojush-Tools · GitHub

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
#import math
import os
from sys import maxint

# Set these before running:
#outputDirectory = "Results/kata-gsxover/lexicase/"
#outputDirectory = "Results/ULTRA/lexicase/"
#outputDirectory = "Results/ULTRA/larger-lexicase/

#outputDirectory = "Results/GECCO13/bio-normal/"
#outputDirectory = "Results/GECCO13/bio-ultra/"

#outputDirectory = "Results/GECCO13/pagie-normal/"
#outputDirectory = "Results/GECCO13/pagie-ultra/"
#outputDirectory = "Results/GECCO13/pagie-no-erc-normal/"
#outputDirectory = "Results/GECCO13/pagie-no-erc-ultra/"
#outputDirectory = "Results/GECCO13/pagie-hogeweg-no-erc-45/"

#outputDirectory = "Results/GECCO13/mom2/"
#outputDirectory = "Results/GECCO13/mom2-lexicase/"
#outputDirectory = "Results/GECCO13/mom2-ultra/"
#outputDirectory = "Results/GECCO13/mom2-tags/"
#outputDirectory = "Results/GECCO13/mom2-env/"
#outputDirectory = "Results/GECCO13/mom2-lex-ultra/"

#outputDirectory = "Results/GECCO13/mom3/"
#outputDirectory = "Results/GECCO13/mom3-lexicase/"
#outputDirectory = "Results/GECCO13/mom3-ultra/"
#outputDirectory = "Results/GECCO13/mom3-lex-ultra/"

#outputDirectory = "Results/GECCO14/wc/empties-max-points-1000-RETRY/"
#outputDirectory = "Results/GECCO14/wc/padding-max-points-1000/"
#outputDirectory = "Results/GECCO14/wc/bushy-max-points-1000/"

outputDirectory = "Results/plush-testing/dm2-uniform-xover-mut/"

outputFilePrefix = "log"
outputFileSuffix = ".txt"

#outputFilePrefix = "bio-normallog"


# Helper function
def my_mean(list_of_nums):
    if len(list_of_nums) == 0:
        return 0
    return float(sum(list_of_nums)) / float(len(list_of_nums))

# Main area
i = 0

if outputDirectory[-1] != '/':
    outputDirectory += '/'
dirList = os.listdir(outputDirectory)


#print "Computing best hits per generation..."
max_generations = 0
sizes_per_gen = []

while (outputFilePrefix + str(i) + outputFileSuffix) in dirList:
    #sys.stdout.write('.')
    #if i % 50 == 49:
    #    print

    fileName = (outputFilePrefix + str(i) + outputFileSuffix)
    f = open(outputDirectory + fileName)

    gen = 0
    mean_size = 0

    for line in f:
        if i == 0:
            if line.startswith("max-generations"):
                max_generations = int(line.split()[-1])
                sizes_per_gen = [[] for x in range(max_generations + 1)]

        if line.startswith(";; -*- Report"):
            gen = int(line.split()[-1])

        if line.startswith("Average program size in population"):
            mean_size = float(line.split()[-1])
            sizes_per_gen[gen].append(mean_size)

    i += 1


#for i, (gen, size, mean_size)  in enumerate(sizes_per_gen):
#    print "Run: %3i  | Gen: %4i  | Best Prog Size = %4i  | Mean Prog Size = %.1f" % (i, gen, size, mean_size)

for i, mean_sizes_at_gen in enumerate(sizes_per_gen):
    #print len(mean_sizes_at_gen)
    print "Gen: %4i | NumRunsRemaining: %4i | MeanOfMeanProgSize: %.1f" % (i, len(mean_sizes_at_gen), my_mean(mean_sizes_at_gen))