wfolsom/cs6601p2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
dkohl
2012-03-12 21:01:30 -04:00
parent 9235ff8d0f
commit 12b4bab59d
3 changed files with 324 additions and 298 deletions
Download Patch File Download Diff File Expand all files Collapse all files

126
src/dkohl/bayes/builders/FoodNetBuilder.java
View File

@@ -25,7 +25,7 @@ public class FoodNetBuilder {
public static final String TASTE = "Taste";
public static final String SOMEONE_VEGETARIAN = "vegetarian";
//public static final String SOMEONE_ALLERGIC_NUTS = "allergic-nuts";
// public static final String SOMEONE_ALLERGIC_NUTS = "allergic-nuts";
public static final String CONTAINS_MEAT = "Meat";
public static final String CONTAINS_VEGETABLE = "Vegetable";
@@ -33,8 +33,9 @@ public class FoodNetBuilder {
public static final String CONTAINS_PORK = TYPE.PORK.toString();
public static final String CONTAINS_TOMATOS = TYPE.TOMATO.toString();
public static final String CONTAINS_POTATOS = TYPE.POTATO.toString();
//public static final String CONTAINS_NUTS = "Nuts";
//public static final String CONTAINS_GENERIC_NUTS = TYPE.GENERIC_NUTS.toString();
// public static final String CONTAINS_NUTS = "Nuts";
// public static final String CONTAINS_GENERIC_NUTS =
// TYPE.GENERIC_NUTS.toString();
public static final String TRUE_VALUE = "true";
public static final String FALSE_VALUE = "false";
@@ -44,18 +45,24 @@ public class FoodNetBuilder {
public static final String RATING_DOMAIN[] = { "1", "2", "3", "4", "5",
"6", "7", "8", "9", "10" };
private static final String[] VARIABLES = { SOMEONE_VEGETARIAN, /*SOMEONE_ALLERGIC_NUTS,*/
private static final String[] VARIABLES = { SOMEONE_VEGETARIAN, /*
* SOMEONE_ALLERGIC_NUTS
* ,
*/
CONTAINS_BEEF, CONTAINS_MEAT, CONTAINS_PORK, CONTAINS_POTATOS,
CONTAINS_TOMATOS, CONTAINS_VEGETABLE, /*CONTAINS_NUTS, CONTAINS_GENERIC_NUTS,*/ TASTE };
CONTAINS_TOMATOS, CONTAINS_VEGETABLE, /*
* CONTAINS_NUTS,
* CONTAINS_GENERIC_NUTS,
*/TASTE };
private static final String[] OBSERVED = { CONTAINS_BEEF, CONTAINS_PORK,
CONTAINS_POTATOS, CONTAINS_TOMATOS/*, CONTAINS_GENERIC_NUTS*/};
CONTAINS_POTATOS, CONTAINS_TOMATOS /* , CONTAINS_GENERIC_NUTS */};
public static Ontology createOntology() {
HashSet<String> classes = new HashSet<String>();
classes.add(CONTAINS_MEAT);
//classes.add(CONTAINS_NUTS);
// classes.add(CONTAINS_NUTS);
classes.add(CONTAINS_VEGETABLE);
Ontology ontology = new Ontology(classes);
@@ -63,7 +70,7 @@ public class FoodNetBuilder {
ontology.define(CONTAINS_PORK, CONTAINS_MEAT);
ontology.define(CONTAINS_BEEF, CONTAINS_MEAT);
//ontology.define(CONTAINS_GENERIC_NUTS, CONTAINS_NUTS);
// ontology.define(CONTAINS_GENERIC_NUTS, CONTAINS_NUTS);
ontology.define(CONTAINS_TOMATOS, CONTAINS_VEGETABLE);
ontology.define(CONTAINS_POTATOS, CONTAINS_VEGETABLE);
@@ -94,20 +101,24 @@ public class FoodNetBuilder {
return normPoint;
}
public static DataSet getSurveyDataSet(Survey survey, RecipeBook recipeBook, int startIndex, int endIndex) {
public static DataSet getSurveyDataSet(Survey survey,
RecipeBook recipeBook, int startIndex, int endIndex) {
DataSet data = FoodExampleBuilder.examples();
Ontology onto = createOntology();
for (int dinerIndex = 0; dinerIndex < survey.getDinerCount(); dinerIndex++) {
Diner diner = survey.getDiner(dinerIndex);
for (int dishIndex = startIndex; dishIndex < endIndex; dishIndex++) {
data.add(normalize(createDataPoint(recipeBook, survey.getDish(dishIndex), diner.getRating(dishIndex)),onto));
data.add(normalize(
createDataPoint(recipeBook, survey.getDish(dishIndex),
diner.getRating(dishIndex)), onto));
}
}
return data;
}
public static DataPoint createDataPoint(RecipeBook recipeBook, String recipeName, int weight) {
public static DataPoint createDataPoint(RecipeBook recipeBook,
String recipeName, int weight) {
DataPoint point = new DataPoint();
Recipe recipe = recipeBook.getRecipe(recipeName);
@@ -126,9 +137,9 @@ public class FoodNetBuilder {
point.add(build(CONTAINS_TOMATOS, TRUE_VALUE));
}
/*
if (ingredients.contains(TYPE.GENERIC_NUTS)) {
point.add(build(CONTAINS_GENERIC_NUTS, TRUE_VALUE));
}*/
* if (ingredients.contains(TYPE.GENERIC_NUTS)) {
* point.add(build(CONTAINS_GENERIC_NUTS, TRUE_VALUE)); }
*/
point.add(build(TASTE, "" + weight));
@@ -136,31 +147,25 @@ public class FoodNetBuilder {
}
public static ProbabilityDistribution vegi(Survey survey) {
String names[] = { SOMEONE_VEGETARIAN };
String names[] = {CONTAINS_MEAT, SOMEONE_VEGETARIAN};
ProbabilityTable table = new ProbabilityTable(names);
//if (survey.isDiner("vegetarian")) {
// table.setProbabilityForAssignment("true;", new Probability(1));
// table.setProbabilityForAssignment("false;", new Probability(0));
//} else {
table.setProbabilityForAssignment("true;", new Probability(0));
table.setProbabilityForAssignment("false;", new Probability(1));
//}
table.setProbabilityForAssignment("true;true;", new Probability(0));
table.setProbabilityForAssignment("false;true;", new Probability(1));
table.setProbabilityForAssignment("true;false;", new Probability(1));
table.setProbabilityForAssignment("false;false;", new Probability(1));
return table;
}
/*
public static ProbabilityDistribution allergicNuts(Survey survey) {
String names[] = { SOMEONE_ALLERGIC_NUTS };
ProbabilityTable table = new ProbabilityTable(names);
if (survey.isDiner("allergic-nuts")) {
table.setProbabilityForAssignment("true;", new Probability(1));
table.setProbabilityForAssignment("false;", new Probability(0));
} else {
table.setProbabilityForAssignment("true;", new Probability(0));
table.setProbabilityForAssignment("false;", new Probability(1));
}
return table;
}*/
* public static ProbabilityDistribution allergicNuts(Survey survey) {
* String names[] = { SOMEONE_ALLERGIC_NUTS }; ProbabilityTable table = new
* ProbabilityTable(names); if (survey.isDiner("allergic-nuts")) {
* table.setProbabilityForAssignment("true;", new Probability(1));
* table.setProbabilityForAssignment("false;", new Probability(0)); } else {
* table.setProbabilityForAssignment("true;", new Probability(0));
* table.setProbabilityForAssignment("false;", new Probability(1)); } return
* table; }
*/
public static ProbabilityDistribution beef() {
String names[] = { CONTAINS_MEAT, CONTAINS_BEEF };
@@ -175,23 +180,20 @@ public class FoodNetBuilder {
}
public static ProbabilityDistribution meat() {
String names[] = { SOMEONE_VEGETARIAN, CONTAINS_MEAT };
String names[] = { CONTAINS_MEAT };
ProbabilityTable table = new ProbabilityTable(names);
return table;
}
/*
public static ProbabilityDistribution genericNuts() {
String names[] = { CONTAINS_NUTS, CONTAINS_GENERIC_NUTS };
ProbabilityTable table = new ProbabilityTable(names);
return table;
}
public static ProbabilityDistribution nuts() {
String names[] = { SOMEONE_ALLERGIC_NUTS, CONTAINS_NUTS };
ProbabilityTable table = new ProbabilityTable(names);
return table;
}*/
* public static ProbabilityDistribution genericNuts() { String names[] = {
* CONTAINS_NUTS, CONTAINS_GENERIC_NUTS }; ProbabilityTable table = new
* ProbabilityTable(names); return table; }
*
* public static ProbabilityDistribution nuts() { String names[] = {
* SOMEONE_ALLERGIC_NUTS, CONTAINS_NUTS }; ProbabilityTable table = new
* ProbabilityTable(names); return table; }
*/
public static ProbabilityDistribution tomatos() {
String names[] = { CONTAINS_VEGETABLE, CONTAINS_TOMATOS };
@@ -206,42 +208,47 @@ public class FoodNetBuilder {
}
public static ProbabilityDistribution vegetables() {
String names[] = { CONTAINS_VEGETABLE, SOMEONE_VEGETARIAN };
String names[] = { CONTAINS_VEGETABLE };
ProbabilityTable table = new ProbabilityTable(names);
return table;
}
public static ProbabilityDistribution taste() {
String names[] = { TASTE, CONTAINS_BEEF, CONTAINS_PORK,
CONTAINS_POTATOS, CONTAINS_TOMATOS/*, CONTAINS_GENERIC_NUTS*/ };
CONTAINS_POTATOS, CONTAINS_TOMATOS /* , CONTAINS_GENERIC_NUTS */};
ContinousDistribution distribution = new ContinousDistribution(names, 0);
return distribution;
}
public static BayesNet createDishNet(Survey survey, RecipeBook recipeBook, int startIndex, int endIndex) {
public static BayesNet createDishNet(Survey survey, RecipeBook recipeBook,
int startIndex, int endIndex) {
BayesNet net = new BayesNet(VARIABLES);
// net.setDistribution(new Variable(SOMEONE_VEGETARIAN, DOMAIN), vegi(survey));
//net.setDistribution(new Variable(SOMEONE_ALLERGIC_NUTS, DOMAIN), allergicNuts(survey));
// net.setDistribution(new Variable(SOMEONE_VEGETARIAN, DOMAIN),
// vegi(survey));
// net.setDistribution(new Variable(SOMEONE_ALLERGIC_NUTS, DOMAIN),
// allergicNuts(survey));
net.setDistribution(new Variable(CONTAINS_MEAT, DOMAIN), meat());
//net.setDistribution(new Variable(CONTAINS_NUTS, DOMAIN), nuts());
net.setDistribution(new Variable(SOMEONE_VEGETARIAN, DOMAIN), vegi(survey));
// net.setDistribution(new Variable(CONTAINS_NUTS, DOMAIN), nuts());
net.setDistribution(new Variable(SOMEONE_VEGETARIAN, DOMAIN),
vegi(survey));
net.setDistribution(new Variable(CONTAINS_VEGETABLE, DOMAIN),
vegetables());
net.setDistribution(new Variable(CONTAINS_BEEF, DOMAIN), beef());
net.setDistribution(new Variable(CONTAINS_PORK, DOMAIN), pork());
net.setDistribution(new Variable(CONTAINS_POTATOS, DOMAIN), potatos());
net.setDistribution(new Variable(CONTAINS_TOMATOS, DOMAIN), tomatos());
//net.setDistribution(new Variable(CONTAINS_GENERIC_NUTS, DOMAIN), genericNuts());
// net.setDistribution(new Variable(CONTAINS_GENERIC_NUTS, DOMAIN),
// genericNuts());
net.setDistribution(new Variable(TASTE, RATING_DOMAIN), taste());
Ontology ontology = createOntology();
// for (String category : ontology.getClasses()) {
// net.connect(category, SOMEONE_VEGETARIAN);
// /*net.connect(category, SOMEONE_ALLERGIC_NUTS);*/
// }
// for (String category : ontology.getClasses()) {
// net.connect(category, SOMEONE_VEGETARIAN);
// /*net.connect(category, SOMEONE_ALLERGIC_NUTS);*/
// }
net.connect(SOMEONE_VEGETARIAN, CONTAINS_MEAT);
for (String thing : OBSERVED) {
@@ -249,7 +256,8 @@ public class FoodNetBuilder {
net.connect(TASTE, thing);
}
DataSet dataSet = getSurveyDataSet(survey, recipeBook, startIndex, endIndex);
DataSet dataSet = getSurveyDataSet(survey, recipeBook, startIndex,
endIndex);
for (String category : ontology.getClasses()) {
MaximumLikelihoodEstimation.estimate(dataSet, net, category);

3
src/dkohl/bayes/inference/EnumerateAll.java
View File

@@ -99,6 +99,9 @@ public class EnumerateAll {
temp.addAll(assignments);
temp.add(new Assignment(variable, value));
if(net.getNodes().get(variable.getName()).eval(temp) == null) {
System.out.println(variable.getName());
}
// then evaluate this variable
double val = net.getNodes().get(variable.getName()).eval(temp)
.getProbability();

15
src/dkohl/bayes/statistic/DataSet.java
View File

@@ -80,6 +80,20 @@ public class DataSet extends Vector<DataPoint> {
public Probability prob(Assignment query, LinkedList<Assignment> given) {
int matches = 0;
int num_query_given = 0;
if (given.size() == 0) {
for (DataPoint point : this) {
if (point.containsKey(query.getVariable().getName())) {
matches += 1;
if (point.get(query.getVariable().getName()).getValue()
.equals(query.getValue())) {
num_query_given += 1;
}
}
}
return new Probability(num_query_given / ((double) matches));
} else {
for (DataPoint point : this) {
if (match(query.getVariable().getName(), point, given)) {
matches += 1;
@@ -95,5 +109,6 @@ public class DataSet extends Vector<DataPoint> {
return new Probability(num_query_given / ((double) matches));
}
}
}