Search in sources :

Example 6 with Pc

use of edu.cmu.tetrad.search.Pc in project tetrad by cmu-phil.

the class TestPcd method checkWithKnowledge.

/**
 * Presents the input graph to FCI and checks to make sure the output of FCI is equivalent to the given output
 * graph.
 */
private void checkWithKnowledge(String inputGraph, String outputGraph, IKnowledge knowledge) {
    // Set up graph and node objects.
    Graph graph = GraphConverter.convert(inputGraph);
    // Set up search.
    IndependenceTest independence = new IndTestDSep(graph);
    Pc pc = new Pc(independence);
    pc.setKnowledge(knowledge);
    // Run search
    Graph resultGraph = pc.search();
    // Build comparison graph.
    Graph trueGraph = GraphConverter.convert(outputGraph);
    resultGraph = GraphUtils.replaceNodes(resultGraph, trueGraph.getNodes());
    // Do test.
    assertTrue(resultGraph.equals(trueGraph));
}
Also used : IndependenceTest(edu.cmu.tetrad.search.IndependenceTest) IndTestDSep(edu.cmu.tetrad.search.IndTestDSep) Graph(edu.cmu.tetrad.graph.Graph) Pc(edu.cmu.tetrad.search.Pc)

Aggregations

Pc (edu.cmu.tetrad.search.Pc)6 Graph (edu.cmu.tetrad.graph.Graph)3 RandomGraph (edu.cmu.tetrad.algcomparison.graph.RandomGraph)2 SemBicDTest (edu.cmu.tetrad.algcomparison.independence.SemBicDTest)2 SemBicTest (edu.cmu.tetrad.algcomparison.independence.SemBicTest)2 Node (edu.cmu.tetrad.graph.Node)2 Fges (edu.cmu.tetrad.search.Fges)2 Test (org.junit.Test)2 CovarianceMatrix (edu.cmu.tetrad.data.CovarianceMatrix)1 DataSet (edu.cmu.tetrad.data.DataSet)1 ICovarianceMatrix (edu.cmu.tetrad.data.ICovarianceMatrix)1 DagToPag (edu.cmu.tetrad.search.DagToPag)1 GFci (edu.cmu.tetrad.search.GFci)1 IndTestCramerT (edu.cmu.tetrad.search.IndTestCramerT)1 IndTestDSep (edu.cmu.tetrad.search.IndTestDSep)1 IndTestFisherZ (edu.cmu.tetrad.search.IndTestFisherZ)1 IndependenceTest (edu.cmu.tetrad.search.IndependenceTest)1 SemIm (edu.cmu.tetrad.sem.SemIm)1 SemPm (edu.cmu.tetrad.sem.SemPm)1 File (java.io.File)1