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_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))
