def test_max_likelihood_all_methods():
# Sanity check that all algorithms work on very simple model.
all_methods = ['auto', 'bruteforce', 'tree_dp', 'path_dp', 'junction_tree']
model = PairWiseFiniteModel(2, 2)
model.add_interaction(0, 1, np.array([[0, 0], [0, 1]]))
expected_result = np.array([1, 1])
for method in all_methods:
result = model.max_likelihood(algorithm=method)
assert np.allclose(result, expected_result)
def test_get_dfs_result():
model = PairWiseFiniteModel(4, 2)
j = np.ones((2, 2))
model.add_interaction(2, 3, j)
model.add_interaction(2, 1, j)
model.get_dfs_result() # To test cache invalidation.
model.add_interaction(1, 0, j)
dfs_edges = model.get_dfs_result().dfs_edges
assert np.allclose(dfs_edges, np.array([[0, 1], [1, 2], [2, 3]]))
def test_build_from_interactions():
model = PairWiseFiniteModel(10, 5)
j1 = np.random.random(size=(5, 5))
j2 = np.random.random(size=(5, 5))
model.add_interaction(0, 1, j1)
model.add_interaction(1, 0, j1)
model.add_interaction(1, 2, j2)
assert np.allclose(model.field, np.zeros((10, 5)))
assert model.edges == [(0, 1), (1, 2)]
assert np.allclose(model.get_interaction_matrix(0, 1), j1 + j1.T)
assert np.allclose(model.get_interaction_matrix(2, 1), j2.T)
def test_inference_all_methods():
# Sanity check that all algorithms work on very simple model.
all_methods = [
'auto', 'bruteforce', 'mean_field', 'message_passing', 'tree_dp',
'path_dp', 'junction_tree'
]
model = PairWiseFiniteModel(2, 2)
model.add_interaction(0, 1, np.array([[0, 0], [0, 1]]))
m = np.array([2, 1 + np.exp(1)])
expected_result = InferenceResult(np.log(3 + np.exp(1)),
np.array([m, m]) / np.sum(m))
for method in all_methods:
result = model.infer(algorithm=method)
assert_results_close(result,
expected_result,
log_pf_tol=1.0,
mp_mse_tol=0.1)