def test_DefaultKernel(self):
B1 = Binomial([10, 0.2])
N1 = Normal([0.1, 0.01])
N2 = Normal([0.3, N1])
graph = Normal([B1, N2])
Manager = AcceptedParametersManager([graph])
backend = Backend()
kernel = DefaultKernel([N1, N2, B1])
Manager.update_broadcast(backend, [[2, 0.27, 0.097], [3, 0.32, 0.012]],
np.array([1, 1]),
accepted_cov_mats=[[[0.01, 0], [0, 0.01]],
[]])
kernel_parameters = []
for krnl in kernel.kernels:
kernel_parameters.append(
Manager.get_accepted_parameters_bds_values(krnl.models))
Manager.update_kernel_values(backend,
kernel_parameters=kernel_parameters)
rng = np.random.RandomState(1)
perturbed_values_and_models = kernel.update(Manager, 1, rng)
self.assertEqual(perturbed_values_and_models,
[(N1, [0.17443453636632419]),
(N2, [0.25882435863499248]), (B1, [3])])
def test_return_value_Student_T(self):
N1 = Normal([0.1, 0.01])
N2 = Normal([0.3, N1])
graph = Normal([N1, N2])
Manager = AcceptedParametersManager([graph])
backend = Backend()
kernel = JointPerturbationKernel([MultivariateStudentTKernel([N1, N2], df=2)])
Manager.update_broadcast(backend, [[0.4, 0.09], [0.2, 0.008]], np.array([0.5, 0.2]))
kernel_parameters = []
for krnl in kernel.kernels:
kernel_parameters.append(Manager.get_accepted_parameters_bds_values(krnl.models))
Manager.update_kernel_values(backend, kernel_parameters)
mapping, mapping_index = Manager.get_mapping(Manager.model)
covs = [[[1, 0], [0, 1]], []]
Manager.update_broadcast(backend, accepted_cov_mats=covs)
pdf = kernel.pdf(mapping, Manager, Manager.accepted_parameters_bds.value()[1], [0.3, 0.1])
self.assertTrue(isinstance(pdf, float))
def test_return_value(self):
B1 = Binomial([10, 0.2])
N1 = Normal([0.1, 0.01])
N2 = Normal([0.3, N1])
graph = Normal([B1, N2])
Manager = AcceptedParametersManager([graph])
backend = Backend()
kernel = DefaultKernel([N1, N2, B1])
Manager.update_broadcast(backend, [[2, 0.4, 0.09], [3, 0.2, 0.008]], np.array([0.5, 0.2]))
kernel_parameters = []
for krnl in kernel.kernels:
kernel_parameters.append(Manager.get_accepted_parameters_bds_values(krnl.models))
Manager.update_kernel_values(backend, kernel_parameters)
mapping, mapping_index = Manager.get_mapping(Manager.model)
covs = [[[1,0],[0,1]],[]]
Manager.update_broadcast(backend, accepted_cov_mats=covs)
pdf = kernel.pdf(mapping, Manager, 1, [2,0.3,0.1])
self.assertTrue(isinstance(pdf, float))
def test_Student_T(self):
N1 = Normal([0.1, 0.01])
N2 = Normal([0.3, N1])
graph = Normal([N1, N2])
Manager = AcceptedParametersManager([graph])
backend = Backend()
kernel = JointPerturbationKernel([MultivariateStudentTKernel([N1, N2], df=2)])
Manager.update_broadcast(backend, [[0.27, 0.097], [0.32, 0.012]], np.array([1, 1]))
kernel_parameters = []
for krnl in kernel.kernels:
kernel_parameters.append(Manager.get_accepted_parameters_bds_values(krnl.models))
Manager.update_kernel_values(backend, kernel_parameters)
covs = kernel.calculate_cov(Manager)
print(covs)
self.assertTrue(len(covs) == 1)
self.assertTrue(len(covs[0]) == 2)
def test(self):
B1 = Binomial([10, 0.2])
N1 = Normal([0.1, 0.01])
N2 = Normal([0.3, N1])
graph = Normal([B1, N2])
Manager = AcceptedParametersManager([graph])
backend = Backend()
kernel = DefaultKernel([N1, N2, B1])
Manager.update_broadcast(backend, [[2, 0.27, 0.097], [3, 0.32, 0.012]], np.array([1, 1]))
kernel_parameters = []
for krnl in kernel.kernels:
kernel_parameters.append(Manager.get_accepted_parameters_bds_values(krnl.models))
Manager.update_kernel_values(backend, kernel_parameters)
covs = kernel.calculate_cov(Manager)
self.assertTrue(len(covs)==2)
self.assertTrue(len(covs[0])==2)
self.assertTrue(not(covs[1]))
def test_Student_T(self):
N1 = Normal([0.1, 0.01])
N2 = Normal([0.3, N1])
graph = Normal([N1, N2])
Manager = AcceptedParametersManager([graph])
backend = Backend()
kernel = JointPerturbationKernel([MultivariateStudentTKernel([N1, N2], df=2)])
Manager.update_broadcast(backend, [[0.27, 0.097], [0.32, 0.012]], np.array([1, 1]),
accepted_cov_mats=[[[0.01, 0], [0, 0.01]], []])
kernel_parameters = []
for krnl in kernel.kernels:
kernel_parameters.append(
Manager.get_accepted_parameters_bds_values(krnl.models))
Manager.update_kernel_values(backend, kernel_parameters=kernel_parameters)
rng = np.random.RandomState(1)
perturbed_values_and_models = kernel.update(Manager, 1, rng)
print(perturbed_values_and_models)
self.assertEqual(perturbed_values_and_models,
[(N1, [0.2107982411716391]), (N2, [-0.049106838502166614])])
class DrawFromPosterior(InferenceMethod):
model = None
rng = None
n_samples = None
backend = None
n_samples_per_param = None # this needs to be there otherwise it does not instantiate correctly
def __init__(self, root_models, backend, seed=None, discard_too_large_values=True):
self.model = root_models
self.backend = backend
self.rng = np.random.RandomState(seed)
self.discard_too_large_values = discard_too_large_values
# An object managing the bds objects
self.accepted_parameters_manager = AcceptedParametersManager(self.model)
self.n_samples_per_param = 1
def sample(self, journal_file):
journal = Journal.fromFile(journal_file)
accepted_parameters = journal.get_accepted_parameters(-1)
accepted_weights = journal.get_weights(-1)
n_samples = journal.configuration["n_samples"]
self.accepted_parameters_manager.broadcast(self.backend, 1)
# Broadcast Accepted parameters and Accepted weights
self.accepted_parameters_manager.update_broadcast(self.backend, accepted_parameters=accepted_parameters,
accepted_weights=accepted_weights)
seed_arr = self.rng.randint(0, np.iinfo(np.uint32).max, size=n_samples, dtype=np.uint32)
rng_arr = np.array([np.random.RandomState(seed) for seed in seed_arr])
index_arr = np.arange(0,n_samples,1)
data_arr = []
for i in range(len(rng_arr)):
data_arr.append([rng_arr[i], index_arr[i]])
data_pds = self.backend.parallelize(data_arr)
parameters_simulations_pds = self.backend.map(self._sample_parameter, data_pds)
parameters_simulations = self.backend.collect(parameters_simulations_pds)
parameters, simulations = [list(t) for t in zip(*parameters_simulations)]
parameters = np.squeeze(np.array(parameters))
simulations = np.squeeze(np.array(simulations))
return parameters, simulations
def _sample_parameter(self, data, npc=None):
if isinstance(data, np.ndarray):
data = data.tolist()
rng = data[0]
index = data[1]
rng.seed(rng.randint(np.iinfo(np.uint32).max, dtype=np.uint32))
parameter = self.accepted_parameters_manager.accepted_parameters_bds.value()[index]
print(parameter)
parameter_list = [x[0] for x in parameter]
print(parameter_list)
self.set_parameters(parameter_list)
param = self.get_parameters()
print(param)
y_sim = self.simulate(n_samples_per_param=1)
#y_sim = self.model[0].forward_simulate(parameter_list,1)
return parameter, y_sim