Package ch.idsia.credici.learning
Class BayesianCausalEM
- java.lang.Object
-
- ch.idsia.crema.learning.ExpectationMaximization<ch.idsia.crema.factor.bayesian.BayesianFactor,M>
-
- ch.idsia.crema.learning.DiscreteEM<BayesianCausalEM>
-
- ch.idsia.credici.learning.BayesianCausalEM
-
public class BayesianCausalEM extends ch.idsia.crema.learning.DiscreteEM<BayesianCausalEM>
-
-
Field Summary
Fields Modifier and Type Field Description HashMap<Set<Integer>,ch.idsia.crema.factor.bayesian.BayesianFactor>
targetGenDist
-
Constructor Summary
Constructors Constructor Description BayesianCausalEM(StructuralCausalModel model)
BayesianCausalEM(StructuralCausalModel model, int[] elimSeq)
BayesianCausalEM(StructuralCausalModel model, ch.idsia.crema.inference.JoinInference<ch.idsia.crema.factor.bayesian.BayesianFactor,ch.idsia.crema.factor.bayesian.BayesianFactor> inferenceEngine)
-
Method Summary
Modifier and Type Method Description Collection
getJointProbs(StructuralCausalModel model)
StructuralCausalModel
getPosterior()
StructuralCausalModel
getPrior()
double
getRegularization()
static void
main(String[] args)
BayesianCausalEM
setRegularization(double regularization)
BayesianCausalEM
setTargetGenDist(HashMap<Set<Integer>,ch.idsia.crema.factor.bayesian.BayesianFactor> targetGenDist)
BayesianCausalEM
setTrainableVars(int[] trainableVars)
protected void
stepPrivate(Collection stepArgs)
BayesianCausalEM
usePosteriorCache(boolean active)
-
Methods inherited from class ch.idsia.crema.learning.ExpectationMaximization
addIntermediateModels, getInferenceEngine, getIntermediateModels, getKlthreshold, getPerformedIterations, getTrainableVars, isRecordIntermediate, isStopAtConvergence, isVerbose, run, setInferenceEngine, setInline, setKlthreshold, setPerformedIterations, setRecordIntermediate, setStopAtConvergence, setUpdated, setVerbose, step
-
-
-
-
Constructor Detail
-
BayesianCausalEM
public BayesianCausalEM(StructuralCausalModel model, ch.idsia.crema.inference.JoinInference<ch.idsia.crema.factor.bayesian.BayesianFactor,ch.idsia.crema.factor.bayesian.BayesianFactor> inferenceEngine)
-
BayesianCausalEM
public BayesianCausalEM(StructuralCausalModel model, int[] elimSeq)
-
BayesianCausalEM
public BayesianCausalEM(StructuralCausalModel model)
-
-
Method Detail
-
stepPrivate
protected void stepPrivate(Collection stepArgs) throws InterruptedException
- Specified by:
stepPrivate
in classch.idsia.crema.learning.ExpectationMaximization<ch.idsia.crema.factor.bayesian.BayesianFactor,BayesianCausalEM>
- Throws:
InterruptedException
-
getJointProbs
public Collection getJointProbs(StructuralCausalModel model) throws InterruptedException
- Throws:
InterruptedException
-
setRegularization
public BayesianCausalEM setRegularization(double regularization)
-
getRegularization
public double getRegularization()
-
setTrainableVars
public BayesianCausalEM setTrainableVars(int[] trainableVars)
- Overrides:
setTrainableVars
in classch.idsia.crema.learning.ExpectationMaximization<ch.idsia.crema.factor.bayesian.BayesianFactor,BayesianCausalEM>
-
usePosteriorCache
public BayesianCausalEM usePosteriorCache(boolean active)
-
getPrior
public StructuralCausalModel getPrior()
-
getPosterior
public StructuralCausalModel getPosterior()
- Overrides:
getPosterior
in classch.idsia.crema.learning.ExpectationMaximization<ch.idsia.crema.factor.bayesian.BayesianFactor,BayesianCausalEM>
-
setTargetGenDist
public BayesianCausalEM setTargetGenDist(HashMap<Set<Integer>,ch.idsia.crema.factor.bayesian.BayesianFactor> targetGenDist)
-
main
public static void main(String[] args) throws InterruptedException
- Throws:
InterruptedException
-
-