Examples with DefaultExpressionFactor com.sri.ai.praise.core.representation.interfacebased.factor.core.expression.core.DefaultExpressionFactor used on opensource projects

Example 16 with DefaultExpressionFactor

use of com.sri.ai.praise.core.representation.interfacebased.factor.core.expression.core.DefaultExpressionFactor in project aic-praise by aic-sri-international.

the class TestCases method expressionFactorIsingModel.

private static ArrayList<ExpressionFactor> expressionFactorIsingModel(int gridSize, BiFunction<Pair<Integer, Integer>, Pair<Integer, Integer>, ArrayList<Double>> pairwiseFactorentries, Function<Pair<Integer, Integer>, ArrayList<Double>> singleVariableFactorEntries) {
    ArrayList<ArrayList<ExpressionVariable>> variables = new ArrayList<>();
    Context context = new TrueContext(new CommonTheory());
    for (int i = 0; i < gridSize; i++) {
        ArrayList<ExpressionVariable> col = new ArrayList<>();
        variables.add(col);
        for (int j = 0; j < gridSize; j++) {
            ExpressionVariable v = new DefaultExpressionVariable(makeSymbol("A_" + i + "_" + j));
            col.add(j, v);
            makeSymbol("A_" + i + "_" + j);
            context = context.extendWithSymbolsAndTypes(v, parse("Boolean"));
        }
    }
    ArrayList<ExpressionFactor> result = new ArrayList<>();
    for (int i = 0; i < gridSize - 1; i++) {
        for (int j = 0; j < gridSize; j++) {
            // Expression entry1 = Expressions.makeSymbol(randomGenerator.nextInt(10000)*1./10000.);
            ArrayList<Double> entryList = pairwiseFactorentries.apply(new Pair<>(i, j), new Pair<>(i + 1, j));
            Expression entry1 = makeSymbol(entryList.get(0));
            Expression entry2 = makeSymbol(entryList.get(1));
            Expression entry3 = makeSymbol(entryList.get(2));
            Expression entry4 = makeSymbol(entryList.get(3));
            Expression X = variables.get(i).get(j);
            Expression Y = variables.get(i + 1).get(j);
            result.add(new DefaultExpressionFactor(apply(IF_THEN_ELSE, apply(EQUAL, X, makeSymbol("true")), apply(IF_THEN_ELSE, apply(EQUAL, Y, makeSymbol("true")), entry1, entry2), apply(IF_THEN_ELSE, apply(EQUAL, Y, makeSymbol("true")), entry3, entry4)), context));
        }
    }
    for (int i = 0; i < gridSize; i++) {
        for (int j = 0; j < gridSize - 1; j++) {
            ArrayList<Double> entryList = pairwiseFactorentries.apply(new Pair<>(i, j), new Pair<>(i, j + 1));
            Expression entry1 = makeSymbol(entryList.get(0));
            Expression entry2 = makeSymbol(entryList.get(1));
            Expression entry3 = makeSymbol(entryList.get(2));
            Expression entry4 = makeSymbol(entryList.get(3));
            Expression X = variables.get(i).get(j);
            Expression Y = variables.get(i).get(j + 1);
            result.add(new DefaultExpressionFactor(apply(IF_THEN_ELSE, apply(EQUAL, X, makeSymbol("true")), apply(IF_THEN_ELSE, apply(EQUAL, Y, makeSymbol("true")), entry1, entry2), apply(IF_THEN_ELSE, apply(EQUAL, Y, makeSymbol("true")), entry3, entry4)), context));
        }
    }
    if (!(singleVariableFactorEntries.apply(new Pair<>(0, 0)) == null)) {
        for (int i = 0; i < gridSize; i++) {
            for (int j = 0; j < gridSize; j++) {
                ArrayList<Double> entryList = singleVariableFactorEntries.apply(new Pair<>(i, j));
                Expression entry1 = makeSymbol(entryList.get(0));
                Expression entry2 = makeSymbol(entryList.get(1));
                Expression X = variables.get(i).get(j);
                result.add(new DefaultExpressionFactor(apply(IF_THEN_ELSE, apply(EQUAL, X, makeSymbol("true")), entry1, entry2), context));
            }
        }
    }
    return result;
}