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 CollectiongetJointProbs(StructuralCausalModel model)StructuralCausalModelgetPosterior()StructuralCausalModelgetPrior()doublegetRegularization()static voidmain(String[] args)BayesianCausalEMsetRegularization(double regularization)BayesianCausalEMsetTargetGenDist(HashMap<Set<Integer>,ch.idsia.crema.factor.bayesian.BayesianFactor> targetGenDist)BayesianCausalEMsetTrainableVars(int[] trainableVars)protected voidstepPrivate(Collection stepArgs)BayesianCausalEMusePosteriorCache(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:
stepPrivatein 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:
setTrainableVarsin 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:
getPosteriorin 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
-
-