Added Daniel's Bayes Net code. Converted example code to unit tests. Minor code clean-up.

This commit is contained in:
Woody Folsom
2012-03-11 10:33:45 -04:00
parent a021dc2fc0
commit 571d0a1922
27 changed files with 2310 additions and 0 deletions

View 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());
}
}