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
|
#!/usr/bin/env python
#
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
# Copyright (c) 1997-2016 California Institute of Technology.
# Copyright (c) 2016-2024 The Uncertainty Quantification Foundation.
# License: 3-clause BSD. The full license text is available at:
# - https://github.com/uqfoundation/mystic/blob/master/LICENSE
"""
Same as test_ffit.py
but uses DifferentialEvolutionSolver2 instead.
Note:
1. MPIDifferentialEvolutionSolver is functionally identical to DifferentialEvolultionSolver2.
2. In DifferentialEvolutionSolver, as each trial vector is compared to its target, once the trial beats
the target it enters the generation, replacing the old vector and immediately becoming available
as candidates for creating difference vectors, and for mutations, etc.
"""
from test_ffit import *
def main():
from mystic.solvers import DifferentialEvolutionSolver2
#from pathos.pools import ProcessPool as Pool
from pathos.pools import ParallelPool as Pool
solver = DifferentialEvolutionSolver2(ND, NP)
solver.SetMapper(Pool().map)
solver.SetRandomInitialPoints(min = [-100.0]*ND, max = [100.0]*ND)
solver.SetEvaluationLimits(generations=MAX_GENERATIONS)
solver.SetGenerationMonitor(VerboseMonitor(30))
solver.enable_signal_handler()
strategy = Best1Exp
#strategy = Best1Bin
solver.Solve(ChebyshevCost, termination=VTR(0.01), strategy=strategy, \
CrossProbability=1.0, ScalingFactor=0.9 , \
sigint_callback=plot_solution)
solution = solver.Solution()
return solution
if __name__ == '__main__':
from pathos.helpers import freeze_support, shutdown
freeze_support() # help Windows use multiprocessing
solution = main()
shutdown() # help multiprocessing shutdown all workers
print_solution(solution)
plot_solution(solution)
# end of file
|
