fyp/study.py at master · blake01/fyp · 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
95
96
97
98
99
import openpyxl
import string
import testFunctions
from openpyxl.cell import get_column_letter
from deBase import DERand1Bin, DECurrentToPBest1Bin
from jade import JADEWithArchive as JADE
from sade import SaDE
from jde import jDE
from hybrids import LocalJADE, HybridJADE, HybridjDE, sadJADE
from mixins import LoggingMixin, ParallelCostMixin, ValueToReachMixin
import time
import os
import csv
import sys
import multiprocessing

"""
This file tests the various options of the DifferentialEvolution class by
running them repeatedly on the test functions defined in testFunctions folder.

The results are exported to Microsoft Excel.
"""

ALPHABET = string.uppercase


def study():
    if "--file" in sys.argv:
        f = open('study.out', 'w')
    algorithms = [DERand1Bin, jDE, SaDE, JADE]
    repeats = int(sys.argv[1])
    # Initialise Excel workbook
    wb = openpyxl.Workbook()
    wb_name = 'DE_Tests_%s.xlsx'%(time.strftime('%d-%m-%Y__%H:%M'))
    worksheets = {}
    # Run the tests
    problems = {
        'f1-10d': testFunctions.sphere10d,
        'f1-30d': testFunctions.sphere30d,
        'f2-10d': testFunctions.rosenbrock10d,
        'f2-30d': testFunctions.rosenbrock30d,
        'f3-10d': testFunctions.step10d,
        'f3-30d': testFunctions.step30d,
        'f4-10d': testFunctions.ackley10d,
        'f4-30d': testFunctions.ackley30d,
        'f5-5d': testFunctions.shekel5d,
    }
    if "--problem" in sys.argv:
        i = sys.argv.index("--problem")
        p = sys.argv[i+1]
        problems = {p: problems[p]}
    for problem_descr, problem in sorted(problems.iteritems()):
        print 'Testing %s'%(problem_descr)
        for Algorithm in algorithms:
            problem_id = '%s_%s'%(problem_descr, Algorithm.__name__)
            class DE(ValueToReachMixin, LoggingMixin, Algorithm):
                pass
            if '30d' in problem_id:
                mfe = 100000
                if "--parallel" in sys.argv:
                    class DE(ParallelCostMixin, ValueToReachMixin, LoggingMixin, Algorithm):
                        pass
                    print '\nParallel Processing on %s CPUs\n'%(multiprocessing.cpu_count())
                    if "--file" in sys.argv:
                        f.write('\nParallel Processing on %s CPUs\n\n'%(multiprocessing.cpu_count()))
            else:
                mfe = 50000
            worksheets[problem_id] = wb.create_sheet()
            ws = worksheets[problem_id]
            ws.title = problem_id
            failures = 0
            for i in xrange(repeats):
                # Run the optimisation
                uuid = str(problem_id) + str(i)
                de = DE(costFile=problem, uuid=uuid, valueToReach=1e-6, maxFunctionEvals=mfe)
                run_name = '- Run %s with %s'%(i+1, problem_id)
                print run_name
                bestVector = de.optimise()
                print bestVector
                if "--file" in sys.argv:
                    f.write('%s\n%s\n\n'%(run_name, bestVector))
                # Dump convergence history
                with open(uuid + '.csv', 'rb') as csvfile:
                    reader = csv.reader(csvfile)
                    for row_index, row in enumerate(reader):
                        for column_index, cell in enumerate(row):
                            column_letter = get_column_letter((4 * i) + (column_index + 1))
                            ws.cell('%s%s'%(column_letter, (row_index + 1))).value = cell
                os.remove(uuid + '.csv')
                del de

    print 'Writing results to Excel...'
    # Remove the default sheet
    ws = wb.get_sheet_by_name('Sheet')
    wb.remove_sheet(ws)
    wb.save(wb_name)

if __name__ == '__main__':
    study()