use of com.sri.ai.praise.core.representation.classbased.expressionbased.core.DefaultExpressionBasedProblem in project aic-praise by aic-sri-international.
the class ExpressionBasedProblemTest method runTest.
private void runTest(String modelString, String queryString, String expectedFactorExpressionsIncludingQueryDefinitionIfAnyString, String expectedQuerySymbolString, boolean expectedQueryIsCompound, String expectedSymbolsAndTypesString) {
HOGMExpressionBasedModel model = new HOGMExpressionBasedModel(modelString);
Expression queryExpression = parse(queryString);
ExpressionBasedProblem problem = new DefaultExpressionBasedProblem(queryExpression, model);
String actualFactorExpressionsIncludingQueryDefinitionIfAnyString = problem.getFactorExpressionsIncludingQueryDefinitionIfAny().toString();
println(expectedFactorExpressionsIncludingQueryDefinitionIfAnyString);
println(actualFactorExpressionsIncludingQueryDefinitionIfAnyString);
assertEquals(expectedFactorExpressionsIncludingQueryDefinitionIfAnyString, actualFactorExpressionsIncludingQueryDefinitionIfAnyString);
String actualQuerySymbol = problem.getQuerySymbol().toString();
println("expected querySymbol: " + expectedQuerySymbolString);
println("actual querySymbol: " + actualQuerySymbol);
assertEquals(expectedQuerySymbolString, actualQuerySymbol);
boolean actualQueryIsCompound = problem.getQueryIsCompound();
println("expected queryIsCompound: " + expectedQueryIsCompound);
println("actual queryIsCompound: " + actualQueryIsCompound);
assertEquals(expectedQueryIsCompound, actualQueryIsCompound);
String actualSymbolsAndTypesString = problem.getContext().getSymbolsAndTypes().toString();
println("expected context string: " + expectedSymbolsAndTypesString);
println("actual context string: " + actualSymbolsAndTypesString);
assertEquals(expectedSymbolsAndTypesString, actualSymbolsAndTypesString);
}
use of com.sri.ai.praise.core.representation.classbased.expressionbased.core.DefaultExpressionBasedProblem in project aic-praise by aic-sri-international.
the class ExpressionBasedSolverTest method testAPI.
@Test
public void testAPI() {
// IMPORTANT: this test is reproduced in the User Guide as an example,
// so it should be kept in sync with it.
String modelString = "" + "random earthquake: Boolean;" + "random burglary: Boolean;" + "random alarm: Boolean;" + "" + "earthquake 0.01;" + "burglary 0.1;" + "" + "if earthquake" + " then if burglary" + " then alarm 0.95" + " else alarm 0.6" + " else if burglary" + " then alarm 0.9" + " else alarm 0.01;" + " " + "not alarm;" + "";
Expression evidence = parse("not alarm");
// can be any boolean expression
boolean exploitFactorization = true;
// exploit factorization (that is, employ ExpressionVariable Elimination,
// as opposed to summing over the entire joint probability distribution).
DefaultExpressionBasedProblem problem;
Expression marginal;
HOGMExpressionBasedModel model = new HOGMExpressionBasedModel(modelString);
model = model.getConditionedModel(evidence);
Expression queryExpression;
ExpressionBasedSolver solver = new EvaluationExpressionBasedSolver(exploitFactorization);
queryExpression = parse("not earthquake");
// can be any boolean expression, or any random variable
problem = new DefaultExpressionBasedProblem(queryExpression, model);
marginal = solver.solve(problem);
System.out.println("Marginal is " + marginal);
queryExpression = parse("earthquake");
problem = new DefaultExpressionBasedProblem(queryExpression, model);
marginal = solver.solve(problem);
System.out.println("Marginal is " + marginal);
}
use of com.sri.ai.praise.core.representation.classbased.expressionbased.core.DefaultExpressionBasedProblem in project aic-praise by aic-sri-international.
the class ExpressionBasedSolver method solve.
default Expression solve(Expression queryExpression, ExpressionBasedModel expressionBasedModel) {
ExpressionBasedProblem problem = new DefaultExpressionBasedProblem(queryExpression, expressionBasedModel);
Expression result = solve(problem);
return result;
}
use of com.sri.ai.praise.core.representation.classbased.expressionbased.core.DefaultExpressionBasedProblem in project aic-praise by aic-sri-international.
the class ExpressionBasedSolverTest method runTest.
private void runTest(Expression queryExpression, ExpressionBasedModel model, Expression expected, boolean useFactorization) throws AssertionError {
ExpressionBasedProblem problem = new DefaultExpressionBasedProblem(queryExpression, model);
ExpressionBasedSolver[] solvers = new ExpressionBasedSolver[] { // new EvaluationExpressionBasedSolver(useFactorization),
new ExactBPExpressionBasedSolver() };
for (ExpressionBasedSolver solver : solvers) {
Expression marginal = solver.solve(problem);
checkResult(problem, expected, marginal, solver);
}
}
Aggregations