# ------------------------------------------------------------------------ # simulation REPEATS = 20 # Number of trials T = 1000 * 60 # Duration of each trial if __name__ == '__main__': rewire_probs = rn.uniform(0,1,REPEATS) mean_firings_res = [] try: pool = Pool(8) function_arg = itertools.izip(itertools.repeat(T),rewire_probs,itertools.repeat(2),itertools.repeat(1000), itertools.repeat(False)) mean_firings_res = pool.map(simulation_wrapper_star, function_arg) except: for p in np.nditer(rewire_probs): res = simulation_wrapper(T,p,2,1000,False) mean_firings_res.append(res) # ------------------------------------------------------------------------ # JIDT integration calculation #Start JVM startJVM(getDefaultJVMPath(), "-Djava.class.path=" + "infodynamics.jar") teCalcClass = JPackage("infodynamics.measures.continuous.kraskov").MultiInfoCalculatorKraskov2 teCalc = teCalcClass() integration_result = [] for [mean_firings, p] in mean_firings_res: print 'Calculating multi-information for p = ' + str(p) teCalc.initialise(8) teCalc.startAddObservations()
from Izhikevich import ConnectIzhikevichNetworkLayers,GenerateNetwork, IzhikevichModularNetwork, RewireModularNetwork from Run import RunSimulation,simulation_wrapper_star,simulation_wrapper import matplotlib.pyplot as plt import numpy as np import numpy.random as rn # try to import modules for multiprocessing try: from multiprocessing import Pool import itertools except: pass # ------------------------------------------------------------------------ # simulation T = 1000 if __name__ == '__main__': rewire_probs = np.arange(0,0.6,0.1) mean_firings_res = [] try: pool = Pool(8) function_arg = itertools.izip(itertools.repeat(T),rewire_probs,itertools.repeat(1),itertools.repeat(0), itertools.repeat(True)) mean_firings_res = pool.map(simulation_wrapper_star, function_arg) except: for p in np.nditer(rewire_probs): res = simulation_wrapper(T,p,1,0,True) mean_firings_res.append(res)