Package edu.cmu.tetrad.bayes

Class DirichletDataSetProbs

java.lang.Object

edu.cmu.tetrad.bayes.DirichletDataSetProbs

public final class DirichletDataSetProbs
extends Object

Estimates probabilities directly from data on the fly using maximum likelihood method.

Version:

$Id: $Id

Author:

josephramsey

Constructor Summary

Constructors

Constructor	Description
DirichletDataSetProbs(DataSet dataSet, double symmValue)	Creates a cell count table for the given data set.

Method Summary

Modifier and Type	Method	Description
double	getCellProb(int[] variableValues)	getCellProb.
double	getConditionalProb(Proposition assertion, Proposition condition)	Calculates the conditional probability of an assertion given a condition in the context of a Bayes IM.
DataSet	getDataSet()	Getter for the field dataSet.
double	getProb(Proposition assertion)	Retrieves the probability of the given assertion in the DirichletDataSetProbs.
List<Node>	getVariables()	getVariables.
boolean	isBoundsEnforced()	True iff bounds checking is performed on variable values indices.
void	setBoundsEnforced(boolean boundsEnforced)	True iff bounds checking is performed on variable values indices.

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

DirichletDataSetProbs

public DirichletDataSetProbs(DataSet dataSet,
 double symmValue)

Creates a cell count table for the given data set.

Parameters:

dataSet - a DataSet object

symmValue - a double

Method Details

getCellProb

public double getCellProb(int[] variableValues)

getCellProb.

Parameters:

variableValues - an array of objects

Returns:

the estimated probability for the given cell. The order of the variable values is the order of the variables in getVariable().

getProb

public double getProb(Proposition assertion)

Retrieves the probability of the given assertion in the DirichletDataSetProbs.

Parameters:

assertion - The proposition to be checked for probability.

Returns:

The probability of the assertion as a double value.

getConditionalProb

public double getConditionalProb(Proposition assertion,
 Proposition condition)

Calculates the conditional probability of an assertion given a condition in the context of a Bayes IM.

Parameters:

assertion - the proposition representing the assertion

condition - the proposition representing the condition

Returns:

the conditional probability as a double value

Throws:

IllegalArgumentException - if the assertion and condition are not for the same Bayes IM or if the assertion variable and data variables are different or in a different order

getDataSet

public DataSet getDataSet()

Getter for the field dataSet.

Returns:

the dataset that this is estimating probabilities for.

getVariables

public List<Node> getVariables()

getVariables.

Returns:

the list of variables for the dataset that this is estimating probabilities for.

isBoundsEnforced

public boolean isBoundsEnforced()

True iff bounds checking is performed on variable values indices.

Returns:

a boolean

setBoundsEnforced

public void setBoundsEnforced(boolean boundsEnforced)

True iff bounds checking is performed on variable values indices.

Parameters:

boundsEnforced - a boolean