+ eda algo: same algo than eda-sa without sa, + plotting scripts and problem functions moved to application/common

This commit is contained in:
Caner Candan 2010-09-22 14:38:15 +02:00
commit 65191e2212
16 changed files with 595 additions and 14 deletions

View file

@ -0,0 +1,14 @@
PROJECT(common)
SET(RESOURCES
gplot.py
ggobi.py
)
FOREACH(file ${RESOURCES})
EXECUTE_PROCESS(
COMMAND ${CMAKE_COMMAND} -E copy_if_different
${CMAKE_CURRENT_SOURCE_DIR}/${file}
${DO_BINARY_DIR}/${file}
)
ENDFOREACH(file)

View file

@ -0,0 +1,42 @@
#ifndef _Rosenbrock_h
#define _Rosenbrock_h
#include <eo>
#include <es.h>
#include <es/eoRealInitBounded.h>
#include <es/eoRealOp.h>
#include <es/eoEsChromInit.h>
#include <es/eoRealOp.h>
#include <es/make_real.h>
#include <apply.h>
#include <eoProportionalCombinedOp.h>
template < typename EOT >
class Rosenbrock : public eoEvalFunc< EOT >
{
public:
typedef typename EOT::AtomType AtomType;
virtual void operator()( EOT& p )
{
if (!p.invalid())
return;
p.fitness( _evaluate( p ) );
}
private:
AtomType _evaluate( EOT& p )
{
AtomType r = 0.0;
for (unsigned int i = 0; i < p.size() - 1; ++i)
{
r += p[i] * p[i];
}
return r;
}
};
#endif // !_Rosenbrock_h

View file

@ -0,0 +1,42 @@
#ifndef _Sphere_h
#define _Sphere_h
#include <eo>
#include <es.h>
#include <es/eoRealInitBounded.h>
#include <es/eoRealOp.h>
#include <es/eoEsChromInit.h>
#include <es/eoRealOp.h>
#include <es/make_real.h>
#include <apply.h>
#include <eoProportionalCombinedOp.h>
template < typename EOT >
class Sphere : public eoEvalFunc< EOT >
{
public:
typedef typename EOT::AtomType AtomType;
virtual void operator()( EOT& p )
{
if (!p.invalid())
return;
p.fitness( _evaluate( p ) );
}
private:
AtomType _evaluate( EOT& p )
{
AtomType r = 0.0;
for (unsigned int i = 0; i < p.size() - 1; ++i)
{
r += p[i] * p[i];
}
return r;
}
};
#endif // !_Sphere_h

68
application/common/ggobi.py Executable file
View file

@ -0,0 +1,68 @@
#!/usr/bin/env python
from pprint import *
import sys, os
if __name__ == '__main__':
# parameter phase
if len(sys.argv) < 2:
print 'Usage: %s [FILE]' % sys.argv[0]
sys.exit()
filename = sys.argv[1]
lines = open(filename).readlines()
# formatting phase
try:
results = [ x.split() for x in lines[1:-1] ]
except IOError, e:
print 'Error: %s' % e
sys.exit()
# dimension estimating phase
popsize = int(lines[0].split()[0])
dimsize = int(results[0][1])
# printing phase
print 'popsize: %d' % popsize
print 'dimsize: %d' % dimsize
print
pprint( results )
# cvs converting phase
i = 1
for x in results:
x.insert(0, '"%d"' % i)
i += 1
header = ['""', '"fitness"', '"dimsize"']
for i in range(0, dimsize):
header.append( '"dim%d"' % i )
results.insert(0, header)
# cvs printing phase
file_results = '\n'.join( [ ','.join( x ) for x in results ] )
print
print file_results
try:
open('%s.csv' % filename, 'w').write(file_results + '\n')
except IOError, e:
print 'Error: %s' % e
sys.exit()
# ggobi plotting phase
os.system('ggobi %s.csv' % filename)

399
application/common/gplot.py Executable file
View file

@ -0,0 +1,399 @@
#!/usr/bin/env python
"""plot.py -- Plot EDA-SA results file"""
import os, time, math, tempfile
import numpy
try:
import Gnuplot, Gnuplot.PlotItems, Gnuplot.funcutils
except ImportError:
# kludge in case Gnuplot hasn't been installed as a module yet:
import __init__
Gnuplot = __init__
import PlotItems
Gnuplot.PlotItems = PlotItems
import funcutils
Gnuplot.funcutils = funcutils
import optparse, logging, sys
LEVELS = {'debug': logging.DEBUG,
'info': logging.INFO,
'warning': logging.WARNING,
'error': logging.ERROR,
'critical': logging.CRITICAL}
def logger(level_name, filename='plot.log'):
logging.basicConfig(
level=logging.DEBUG,
format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s',
filename=filename, filemode='a'
)
console = logging.StreamHandler()
console.setLevel(LEVELS.get(level_name, logging.NOTSET))
console.setFormatter(logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s'))
logging.getLogger('').addHandler(console)
def parser(parser=optparse.OptionParser()):
parser.add_option('-v', '--verbose', choices=LEVELS.keys(), default='warning', help='set a verbose level')
parser.add_option('-f', '--files', help='give some input sample files separated by comma (cf. gen1,gen2,...)', default='')
parser.add_option('-r', '--respop', help='define the population results containing folder', default='./ResPop')
parser.add_option('-o', '--output', help='give an output filename for logging', default='plot.log')
parser.add_option('-d', '--dimension', help='give a dimension size', default=2)
parser.add_option('-m', '--multiplot', action="store_true", help='plot all graphics in one window', dest="multiplot", default=True)
parser.add_option('-p', '--plot', action="store_false", help='plot graphics separetly, one by window', dest="multiplot")
parser.add_option('-w', '--windowid', help='give the window id you want to display, 0 means we display all ones, this option should be combined with -p', default=0)
parser.add_option('-G', '--graphicsdirectory', help='give a directory name for graphics, this option should be combined with -u', default='plot')
parser.add_option('-g', '--graphicsprefixname', help='give a prefix name for graphics, this option should be combined with -u', default='plot')
parser.add_option('-t', '--terminal', action="store_true", help='display graphics on gnuplot windows', dest="terminal", default=True)
parser.add_option('-u', '--png', action="store_false", help='display graphics on png files', dest="terminal")
options, args = parser.parse_args()
logger(options.verbose, options.output)
return options
options = parser()
def wait(str=None, prompt='Press return to show results...\n'):
if str is not None:
print str
raw_input(prompt)
def draw2DRect(min=(0,0), max=(1,1), color='black', state=None, g=None):
if g == None: g = Gnuplot.Gnuplot()
if state != None: state.append(g)
xmin, ymin = min
xmax, ymax = max
cmd = 'set arrow from %s,%s to %s,%s nohead lc rgb "%s"'
g(cmd % (xmin, ymin, xmin, ymax, color))
g(cmd % (xmin, ymax, xmax, ymax, color))
g(cmd % (xmax, ymax, xmax, ymin, color))
g(cmd % (xmax, ymin, xmin, ymin, color))
return g
def draw3DRect(min=(0,0,0), max=(1,1,1), state=None, g=None):
if g == None: g = Gnuplot.Gnuplot()
if state != None: state.append(g)
# TODO
return g
def getSortedFiles(path):
assert path != None
if options.files == '':
filelist = os.listdir(path)
filelist.sort()
else:
filelist = options.files.split(',')
checkFileErrors(path, filelist)
return filelist
def checkFileErrors(path, filelist):
for filename in filelist:
for line in open('%s/%s' % (path, filename)):
if '-nan' in line:
logging.warning("checkFileErrors: %s/%s file contains bad value, it is going to be skipped" % (path, filename))
filelist.remove(filename)
break
def plotXPointYFitness(path, fields='3:1', state=None, g=None):
if g == None:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s/%s_%s.png\'' % (options.graphicsdirectory, options.graphicsprefixname, 'plotXPointYFitness'))
if state != None: state.append(g)
g.title('Fitness observation')
g.xlabel('Coordinates')
g.ylabel('Fitness (Quality)')
files=[]
for filename in getSortedFiles(path):
files.append(Gnuplot.File(path + '/' + filename, using=fields,
with_='points',
#title='distribution \'' + filename + '\''
title=""
)
)
if len(files) > 0:
g.plot(*files)
return g
def plotXYPointZFitness(path, fields='4:3:1', state=None, g=None):
if g == None:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s/%s_%s.png\'' % (options.graphicsdirectory, options.graphicsprefixname, 'plotXYPointZFitness'))
if state != None: state.append(g)
g.title('Fitness observation in 3-D')
g.xlabel('x-axes')
g.ylabel('y-axes')
g.zlabel('Fitness (Quality)')
files=[]
for filename in getSortedFiles(path):
files.append(Gnuplot.File(path + '/' + filename, using=fields,
with_='points',
#title='distribution \'' + filename + '\''
title=""
)
)
if len(files) > 0:
g.splot(*files)
return g
def plotXYPoint(path, fields='3:4', state=None, g=None):
if g == None:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s/%s_%s.png\'' % (options.graphicsdirectory, options.graphicsprefixname, 'plotXYPoint'))
if state != None: state.append(g)
g.title('Points observation in 2-D')
g.xlabel('x-axes')
g.ylabel('y-axes')
files=[]
for filename in getSortedFiles(path):
files.append(Gnuplot.File(path + '/' + filename, using=fields,
with_='points',
#title='distribution \'' + filename + '\''
title=""
)
)
if len(files) > 0:
g.plot(*files)
return g
def plotXYZPoint(path, fields='3:4:5', state=None, g=None):
if g == None:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s/%s_%s.png\'' % (options.graphicsdirectory, options.graphicsprefixname, 'plotXYZPoint'))
if state != None: state.append(g)
g.title('Points observation in 3-D')
g.xlabel('x-axes')
g.ylabel('y-axes')
g.zlabel('z-axes')
files=[]
for filename in getSortedFiles(path):
files.append(Gnuplot.File(path + '/' + filename, using=fields,
with_='points',
#title='distribution \'' + filename + '\''
title=""
)
)
if len(files) > 0:
g.splot(*files)
return g
def plotParams(path, field='1', state=None, g=None):
if g == None:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s/%s_%s.png\'' % (options.graphicsdirectory, options.graphicsprefixname, 'plotXYZPoint'))
if state != None: state.append(g)
g.title('Hyper-volume comparaison through all dimensions')
g.xlabel('Iterations')
g.ylabel('Hyper-volume')
g.plot(Gnuplot.File(path, with_='lines', using=field,
title='multivariate distribution narrowing'))
return g
def plot2DRectFromFiles(path, state=None, g=None, plot=True):
if g == None:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s_%s.png\'' % (options.graphicsprefixname, 'plot2DRectFromFiles'))
if state != None: state.append(g)
g.title('Rectangle drawing observation')
g.xlabel('x-axes')
g.ylabel('y-axes')
x1,x2,y1,y2 = 0,0,0,0
colors = ['red', 'orange', 'blue', 'green', 'gold', 'yellow', 'gray']
#colors = open('rgb.txt', 'r').readlines()
colors_size = len(colors)
i = 0 # for color
for filename in getSortedFiles(path):
line = open(path + '/' + filename, 'r').readline()
fields = line.split(' ')
if not fields[0] == '2':
print 'plot2DRectFromFiles: higher than 2 dimensions not possible to draw'
return
xmin,ymin,xmax,ymax = fields[1:5]
#print xmin,ymin,xmax,ymax
cur_color = colors[i % colors_size]
draw2DRect((xmin,ymin), (xmax,ymax), cur_color, g=g)
g('set obj rect from %s,%s to %s,%s back lw 1.0 fc rgb "%s" fillstyle solid 1.00 border -1'
% (xmin,ymin,xmax,ymax,cur_color)
)
if plot:
if float(xmin) < x1: x1 = float(xmin)
if float(ymin) < y1: y1 = float(ymin)
if float(xmax) > x2: x2 = float(xmax)
if float(ymax) > y2: y2 = float(ymax)
#print x1,y1,x2,y2
i += 1
#print x1,y1,x2,y2
if plot:
g.plot('[%s:%s][%s:%s] -9999 notitle' % (x1, x2, y1, y2))
return g
def main():
gstate = []
n = int(options.dimension)
w = int(options.windowid)
r = options.respop
if not options.terminal:
try:
os.mkdir(options.graphicsdirectory)
except OSError:
pass
if options.multiplot:
g = Gnuplot.Gnuplot()
if not options.terminal:
g('set terminal png')
g('set output \'%s/%s_%s.png\'' % (options.graphicsdirectory, options.graphicsprefixname, 'multiplot'))
g('set parametric')
g('set nokey')
g('set noxtic')
g('set noytic')
g('set noztic')
g('set size 1.0, 1.0')
g('set origin 0.0, 0.0')
g('set multiplot')
g('set size 0.5, 0.5')
g('set origin 0.0, 0.5')
if n >= 1:
plotXPointYFitness(r, state=gstate, g=g)
g('set size 0.5, 0.5')
g('set origin 0.0, 0.0')
if n >= 2:
plotXPointYFitness(r, '4:1', state=gstate, g=g)
g('set size 0.5, 0.5')
g('set origin 0.5, 0.5')
if n >= 2:
plotXYPointZFitness(r, state=gstate, g=g)
g('set size 0.5, 0.5')
g('set origin 0.5, 0.0')
if n >= 2:
plotXYPoint(r, state=gstate, g=g)
elif n >= 3:
plotXYZPoint(r, state=gstate, g=g)
g('set nomultiplot')
else:
if n >= 1 and w in [0, 1]:
plotXPointYFitness(r, state=gstate)
if n >= 2 and w in [0, 2]:
plotXPointYFitness(r, '4:1', state=gstate)
if n >= 2 and w in [0, 3]:
plotXYPointZFitness(r, state=gstate)
if n >= 3 and w in [0, 4]:
plotXYZPoint(r, state=gstate)
if n >= 2 and w in [0, 5]:
plotXYPoint(r, state=gstate)
# if n >= 1:
# plotParams('./ResParams.txt', state=gstate)
# if n >= 2:
# plot2DRectFromFiles('./ResBounds', state=gstate)
# plotXYPoint(r, state=gstate)
# g = plot2DRectFromFiles('./ResBounds', state=gstate, plot=False)
# plotXYPoint(r, g=g)
if options.terminal:
wait(prompt='Press return to end the plot.\n')
# when executed, just run main():
if __name__ == '__main__':
logging.debug('### plotting started ###')
main()
logging.debug('### plotting ended ###')