def test_circle_optimization(self):
logging.info(
"\t > Evolution: Testing vanilla optimization of a circle ...")
start = time.time()
def optimize_me(traj):
ind = evolution.getIndividualFromTraj(traj)
computation_result = abs((ind.x**2 + ind.y**2) - 1)
fitness_tuple = (computation_result, )
result_dict = {"result": [computation_result]}
return fitness_tuple, result_dict
pars = ParameterSpace(["x", "y"], [[-5.0, 5.0], [-5.0, 5.0]])
evolution = Evolution(
optimize_me,
pars,
weightList=[-1.0],
POP_INIT_SIZE=8,
POP_SIZE=8,
NGEN=2,
filename="test_circle_optimization.hdf",
)
evolution.run(verbose=False)
end = time.time()
logging.info("\t > Done in {:.2f} s".format(end - start))
def test_nsga2(self):
logging.info("\t > Evolution: Testing ALN single node ...")
start = time.time()
def evaluateSimulation(traj):
rid = traj.id
logging.info("Running run id {}".format(rid))
model = evolution.getModelFromTraj(traj)
model.params["dt"] = 0.2
model.params["duration"] = 2 * 1000.0
model.run()
# -------- fitness evaluation here --------
# example: get dominant frequency of activity
frs, powers = func.getPowerSpectrum(
model.rates_exc[:, -int(1000 / model.params["dt"]):],
model.params["dt"],
)
domfr = frs[np.argmax(powers)]
fitness = abs(domfr - 25) # let's try to find a 25 Hz oscillation
fitness_tuple = ()
fitness_tuple += (fitness, )
# multi objective
fitness_tuple += (fitness, )
return fitness_tuple, model.outputs
alnModel = ALNModel()
# alnModel.run(bold=True)
pars = ParameterSpace(["mue_ext_mean", "mui_ext_mean"],
[[0.0, 4.0], [0.0, 4.0]])
evolution = Evolution(
evaluateSimulation,
pars,
algorithm="nsga2",
model=alnModel,
weightList=[-1.0, 1.0],
POP_INIT_SIZE=4,
POP_SIZE=4,
NGEN=2,
filename="test_nsga2.hdf",
)
evolution.run(verbose=False)
evolution.info(plot=False)
traj = evolution.loadResults()
gens, all_scores = evolution.getScoresDuringEvolution()
# overview of current population
evolution.dfPop
# overview of all past individuals
evolution.dfEvolution
end = time.time()
logging.info("\t > Done in {:.2f} s".format(end - start))
def test_all_crossovers(self):
def evo(traj):
return (1,), {}
pars = ParameterSpace(["x"], [[0.0, 4.0]])
evolution = Evolution(evalFunction=evo, parameterSpace=pars, filename="TestEvolutionCrossover.hdf")
evolution.runInitial()
init_pop = evolution.pop.copy()
# perform crossover methods
ind1, ind2 = init_pop[:2]
du.cxNormDraw_adapt(ind1, ind2, 0.4)
du.cxUniform_adapt(ind1, ind2, 0.4)
du.cxUniform_normDraw_adapt(ind1, ind2, 0.4)
def setUpClass(cls):
pars = ParameterSpace(
["mue_ext_mean", "mui_ext_mean", "b"],
[[0.0, 3.0], [0.0, 3.0], [0.0, 100.0]],
)
evolution = Evolution(
lambda v: v,
pars,
weightList=[1.0],
POP_INIT_SIZE=4,
POP_SIZE=4,
NGEN=2,
filename="TestEvolutinUtils.hdf",
)
cls.evolution = evolution
# fake population
pop = evolution.toolbox.population(n=100)
fitness_length = 3
for i, p in enumerate(pop):
if random.random() < 0.1:
fitnessesResult = [np.nan] * fitness_length
else:
fitnessesResult = np.random.random(fitness_length)
p.id = i
p.fitness.values = fitnessesResult
p.fitness.score = np.nansum(p.fitness.wvalues) / (len(
p.fitness.wvalues))
p.gIdx = 0
cls.pop = pop
cls.evolution.pop = pop
cls.evolution.gIdx = 1
def test_optimize_imports(self):
from neurolib.optimize.evolution import Evolution
evolution = Evolution(evalFunction=(lambda f: f), parameterSpace=self.pars)
from neurolib.optimize.exploration import BoxSearch
search = BoxSearch(evalFunction=(lambda f: f), parameterSpace=self.pars)
def test_multimodel_evolve(self):
fhn_node = MultiModel.init_node(FitzHughNagumoNode())
pars = ParameterSpace(
{
"*noise*sigma": [0.0, 0.5],
"*epsilon*": [0.2, 0.8]
},
allow_star_notation=True)
def evaluateSimulation(traj):
model = evolution.getModelFromTraj(traj)
model.run()
# compute power spectrum
frs, powers = func.getPowerSpectrum(
model.x[:, -int(1000 / model.params["dt"]):],
dt=model.params["dt"])
# find the peak frequency
domfr = frs[np.argmax(powers)]
# fitness evaluation: let's try to find a 25 Hz oscillation
fitness = abs(domfr - 25)
# deap needs a fitness *tuple*!
fitness_tuple = ()
# more fitness values could be added
fitness_tuple += (fitness, )
# we need to return the fitness tuple and the outputs of the model
return fitness_tuple, model.outputs
evolution = Evolution(
evalFunction=evaluateSimulation,
parameterSpace=pars,
model=fhn_node,
weightList=[-1.0],
POP_INIT_SIZE=4,
POP_SIZE=4,
NGEN=2,
filename="test_multimodel.hdf",
)
evolution.run(verbose=False)
evolution.info(plot=False)
evolution.loadResults()
_ = evolution.getScoresDuringEvolution()
evolution.dfPop
evolution.dfEvolution
def test_adaptive(self):
logging.info("\t > Evolution: Testing ALN single node ...")
start = time.time()
def evaluateSimulation(traj):
rid = traj.id
logging.info("Running run id {}".format(rid))
model = evolution.getModelFromTraj(traj)
model.params["dt"] = 0.2
model.params["duration"] = 2 * 1000.0
model.run()
# -------- fitness evaluation here --------
# example: get dominant frequency of activity
frs, powers = func.getPowerSpectrum(
model.rates_exc[:, -int(1000 / model.params["dt"]):],
model.params["dt"],
)
domfr = frs[np.argmax(powers)]
fitness = abs(domfr - 25) # let's try to find a 25 Hz oscillation
fitness_tuple = ()
fitness_tuple += (fitness, )
return fitness_tuple, model.outputs
alnModel = ALNModel()
alnModel.run(bold=True)
pars = ParameterSpace(["mue_ext_mean", "mui_ext_mean"],
[[0.0, 4.0], [0.0, 4.0]])
evolution = Evolution(
evaluateSimulation,
pars,
algorithm="adaptive",
model=alnModel,
weightList=[-1.0],
POP_INIT_SIZE=4,
POP_SIZE=4,
NGEN=2,
filename="test_adaptive.hdf",
)
evolution.run(verbose=False)
evolution.info(plot=False)
_ = evolution.loadResults()
gens, all_scores = evolution.getScoresDuringEvolution()
# save the evolution and reload it from disk
fname = "data/test_saved-evolution.dill"
evolution.saveEvolution(fname=fname)
evolution = evolution.loadEvolution(fname)
# overview of current population
evolution.dfPop
# overview of all past individuals
evolution.dfEvolution
# evolution information
evolution.info(plot=False)
dfPop = evolution.dfPop(outputs=True)
self.assertEqual(len(dfPop), len(evolution.pop))
evolution.dfEvolution()
# methods
evolution.getValidPopulation(evolution.pop)
evolution.getInvalidPopulation(evolution.pop)
# evolution.individualToDict(evolution.pop[0])
end = time.time()
logging.info("\t > Done in {:.2f} s".format(end - start))
