edu.cmu.tetrad.algcomparison.statistic.DefiniteDirectedPathPrecision Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tetrad-lib Show documentation

The newest version!

package edu.cmu.tetrad.algcomparison.statistic;

import edu.cmu.tetrad.data.DataModel;
import edu.cmu.tetrad.graph.*;

import java.io.Serial;
import java.util.List;

/**
 * The bidirected true positives.
 *
 * @author josephramsey
 * @version $Id: $Id
 */
public class DefiniteDirectedPathPrecision implements Statistic {
    @Serial
    private static final long serialVersionUID = 23L;

    /**
     * Initializes a new instance of the DefiniteDirectedPathPrecision class.
     */
    public DefiniteDirectedPathPrecision() {
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public String getAbbreviation() {
        return "DDPP";
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public String getDescription() {
        return "Proportion of DP(X, Y) in est for which DD(X, Y) in CPDAG(true)";
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public double getValue(Graph trueGraph, Graph estGraph, DataModel dataModel) {
        int tp = 0, fp = 0;

        List nodes = trueGraph.getNodes();
        Graph cpdag = GraphTransforms.dagToCpdag(trueGraph);

        GraphUtils.addEdgeSpecializationMarkup(estGraph);

        for (Node x : nodes) {
            for (Node y : nodes) {
                if (x == y) continue;

                Edge e = estGraph.getEdge(x, y);

                if (e != null && e.pointsTowards(y) && e.getProperties().contains(Edge.Property.dd)) {

//                if (estGraph.existsDirectedPathFromTo(x, y)) {
                    if (cpdag.paths().existsDirectedPath(x, y)) {
                        tp++;
                    } else {
                        fp++;
                    }
//                }
                }
            }
        }

        return tp / (double) (tp + fp);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public double getNormValue(double value) {
        return value;
    }
}