Added Daniel's Bayes Net code. Converted example code to unit tests. Minor code clean-up.
This commit is contained in:
39
test/dkohl/bayes/example/AlarmExampleTest.java
Normal file
39
test/dkohl/bayes/example/AlarmExampleTest.java
Normal file
@@ -0,0 +1,39 @@
|
||||
package dkohl.bayes.example;
|
||||
|
||||
import static org.hamcrest.core.IsEqual.equalTo;
|
||||
import static org.junit.Assert.assertThat;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import java.util.LinkedList;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import dkohl.bayes.bayesnet.BayesNet;
|
||||
import dkohl.bayes.example.builders.AlarmNetBuilderTable;
|
||||
import dkohl.bayes.example.builders.AlarmNetBuilderTree;
|
||||
import dkohl.bayes.inference.EnumerateAll;
|
||||
import dkohl.bayes.probability.ProbabilityAssignment;
|
||||
import dkohl.bayes.probability.Variable;
|
||||
|
||||
public class AlarmExampleTest {
|
||||
|
||||
@Test
|
||||
public void testAlarmExample() {
|
||||
BayesNet sprinkler = AlarmNetBuilderTable.alarm();
|
||||
sprinkler = AlarmNetBuilderTree.sprinkler();
|
||||
// P(B | j, m)
|
||||
LinkedList<ProbabilityAssignment> probs = EnumerateAll.enumerateAsk(
|
||||
new Variable(AlarmNetBuilderTable.BURGLARY,
|
||||
AlarmNetBuilderTable.DOMAIN), sprinkler,
|
||||
AlarmNetBuilderTable.completeQueryBulgary());
|
||||
System.out.print("Burglary: <");
|
||||
for (ProbabilityAssignment p : probs) {
|
||||
System.out.print("p: "+ p.toString() + ",");
|
||||
}
|
||||
System.out.println(">");
|
||||
|
||||
//assert that burglary is more likely to be false
|
||||
assertThat(probs.size(),equalTo(2));
|
||||
assertTrue(probs.get(0).getProbability() < probs.get(1).getProbability());
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user