def test_Z3(): """ Test of implementation 2. """ traj = simple_traj10() theta = np.array([500.0, -100.0]) weights = [[0.1, 0.9], [0.9, 0.1]] elts = LearningElementsSecure(traj, theta, weights=weights) elts.computeLogZ()
def test_opt0_traj10_1(): """ test_opt0_traj10_1 """ traj = simple_traj10() start_theta = np.array([-1.0, 2.0]) choices = [0, 0] traj_choices = [(traj, choices)] obj_fun_2 = trajs_obj_fun_2(traj_choices) obj_fun_1 = trajs_obj_fun_1(traj_choices) obj_fun_0bis = trajs_obj_fun_0bis(traj_choices) max_iters = 5 (theta_0bis, ys_0bis) = \ optimize_function(obj_fun_0bis, start_theta, max_iters, regularizer=1e-4) (theta_1, ys_1) = \ optimize_function(obj_fun_1, start_theta, max_iters, regularizer=1e-4) (theta_2, ys_2) = \ optimize_function(obj_fun_2, start_theta, max_iters, regularizer=1e-4) assert(np.abs(theta_1 - theta_0bis).max() < 1e-3), (theta_1, theta_0bis) assert(np.abs(theta_2 - theta_0bis).max() < 1e-3), (theta_2, theta_0bis) assert within(ys_1[0], ys_0bis[0], 1e-3), (theta_0bis, ys_0bis, theta_1, ys_1) assert within(ys_2[0], ys_0bis[0], 1e-3), (theta_0bis, ys_0bis, theta_2, ys_2)
def test_opt0_traj10_1(): """ test_opt0_traj10_1 """ traj = simple_traj10() start_theta = np.array([-1.0, 2.0]) choices = [0, 0] traj_choices = [(traj, choices)] obj_fun_2 = trajs_obj_fun_2(traj_choices) obj_fun_1 = trajs_obj_fun_1(traj_choices) obj_fun_0bis = trajs_obj_fun_0bis(traj_choices) max_iters = 5 (theta_0bis, ys_0bis) = \ optimize_function(obj_fun_0bis, start_theta, max_iters, regularizer=1e-4) (theta_1, ys_1) = \ optimize_function(obj_fun_1, start_theta, max_iters, regularizer=1e-4) (theta_2, ys_2) = \ optimize_function(obj_fun_2, start_theta, max_iters, regularizer=1e-4) assert (np.abs(theta_1 - theta_0bis).max() < 1e-3), (theta_1, theta_0bis) assert (np.abs(theta_2 - theta_0bis).max() < 1e-3), (theta_2, theta_0bis) assert within(ys_1[0], ys_0bis[0], 1e-3), (theta_0bis, ys_0bis, theta_1, ys_1) assert within(ys_2[0], ys_0bis[0], 1e-3), (theta_0bis, ys_0bis, theta_2, ys_2)