com.github.vangj.jbayes.inf.exact.graph.pptc.SepSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jbayes-inference Show documentation

A very cool project for BBN inference using approximate and exact algorithms.

The newest version!

package com.github.vangj.jbayes.inf.exact.graph.pptc;

import com.github.vangj.jbayes.inf.exact.graph.Node;
import com.github.vangj.jbayes.inf.exact.graph.util.NodeUtil;
import java.util.LinkedHashSet;
import java.util.Set;

/**
 * A separation set (which is a sub-type of a clique).
 */
public class SepSet extends Clique {

  private final Clique left;
  private final Clique right;

  public SepSet(Clique left, Clique right) {
    super();
    this.left = left;
    this.right = right;
    Set set1 = new LinkedHashSet<>(left.nodes.values());
    Set set2 = new LinkedHashSet<>(right.nodes.values());
    set1.retainAll(set2);
    set1.forEach(node -> this.nodes.put(node.getId(), node));
  }

  /**
   * Left clique.
   *
   * @return Clique.
   */
  public Clique left() {
    return left;
  }

  /**
   * Right clique.
   *
   * @return Clique.
   */
  public Clique right() {
    return right;
  }

  /**
   * Cost is the weight of the left and right cliques added.
   *
   * @return Cost.
   */
  public int cost() {
    return left.weight() + right.weight();
  }

  /**
   * Mass is the number of nodes in this separation set.
   *
   * @return Mass.
   */
  public int mass() {
    return nodes.size();
  }

  @Override
  public String id() {
    return NodeUtil.id(nodes(), "|", "|");
  }
}