def test_lws_allevidence(self):
self.assertDictEqual(
marginal_lws_a(self.bn,
evidence={
'Burglary': 'Yes',
'Alarm': 'Yes',
'Earthquake': 'Yes',
'JohnCalls': 'Yes',
'MaryCalls': 'Yes'
}), {
'Alarm': {
'No': 0.0,
'Yes': 1.0
},
'Burglary': {
'No': 0.0,
'Yes': 1.0
},
'Earthquake': {
'No': 0.0,
'Yes': 1.0
},
'JohnCalls': {
'No': 0.0,
'Yes': 1.0
},
'MaryCalls': {
'No': 0.0,
'Yes': 1.0
}
})
def test_likelihood_weighted_sample(self):
"""
THIS HAS BEEN VALIDATED AGAINST BNLEARN
"""
np.random.seed(3636)
self.assertDictEqual(
marginal_lws_a(self.bn, evidence={'Burglary': 'Yes'}), {
'Alarm': {
'No': 0.058,
'Yes': 0.942
},
'Burglary': {
'No': 0.0,
'Yes': 1.0
},
'Earthquake': {
'No': 0.997,
'Yes': 0.003
},
'JohnCalls': {
'No': 0.136,
'Yes': 0.864
},
'MaryCalls': {
'No': 0.343,
'Yes': 0.657
}
})
def test_likelihood_weighted_sample(self):
"""
THIS HAS BEEN VALIDATED AGAINST BNLEARN
"""
np.random.seed(3636)
self.assertDictEqual(
marginal_lws_a(self.bn, evidence={"Burglary": "Yes"}),
{
"Alarm": {"No": 0.058, "Yes": 0.942},
"Burglary": {"No": 0.0, "Yes": 1.0},
"Earthquake": {"No": 0.997, "Yes": 0.003},
"JohnCalls": {"No": 0.136, "Yes": 0.864},
"MaryCalls": {"No": 0.343, "Yes": 0.657},
},
)
def test_lws_allevidence(self):
self.assertDictEqual(
marginal_lws_a(
self.bn,
evidence={
"Burglary": "Yes",
"Alarm": "Yes",
"Earthquake": "Yes",
"JohnCalls": "Yes",
"MaryCalls": "Yes",
},
),
{
"Alarm": {"No": 0.0, "Yes": 1.0},
"Burglary": {"No": 0.0, "Yes": 1.0},
"Earthquake": {"No": 0.0, "Yes": 1.0},
"JohnCalls": {"No": 0.0, "Yes": 1.0},
"MaryCalls": {"No": 0.0, "Yes": 1.0},
},
)
