org.integratedmodelling.engine.modelling.bayes.rw.RiskwizBayesianInference Maven / Gradle / Ivy
The newest version!
/*******************************************************************************
* Copyright (C) 2007, 2015:
*
* - Ferdinando Villa
* - integratedmodelling.org
* - any other authors listed in @author annotations
*
* All rights reserved. This file is part of the k.LAB software suite,
* meant to enable modular, collaborative, integrated
* development of interoperable data and model components. For
* details, see http://integratedmodelling.org.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the Affero General Public License
* Version 3 or any later version.
*
* This program is distributed in the hope that it will be useful,
* but without any warranty; without even the implied warranty of
* merchantability or fitness for a particular purpose. See the
* Affero General Public License for more details.
*
* You should have received a copy of the Affero General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
* The license is also available at: https://www.gnu.org/licenses/agpl.html
*******************************************************************************/
package org.integratedmodelling.engine.modelling.bayes.rw;
import java.util.Map;
import org.integratedmodelling.engine.modelling.bayes.IBayesianInference;
import org.integratedmodelling.exceptions.KlabException;
import org.integratedmodelling.exceptions.KlabRuntimeException;
import org.integratedmodelling.riskwiz.bn.BeliefNetwork;
import org.integratedmodelling.riskwiz.inference.ls.JoinTree;
import org.integratedmodelling.riskwiz.inference.ls.JoinTreeCompiler;
import org.integratedmodelling.riskwiz.jtree.JTInference;
public class RiskwizBayesianInference implements IBayesianInference {
JTInference inference = null;
RiskwizBayesianInference(BeliefNetwork bn) {
this.inference = new JTInference();
try {
this.inference.initialize(bn, new JoinTreeCompiler());
} catch (Exception e) {
throw new KlabRuntimeException(e);
}
}
@Override
public void run() {
this.inference.run();
}
@Override
public void setEvidence(String node, String outcome) throws KlabException {
this.inference.setObservation(node, outcome);
}
@Override
public double getMarginal(String node, String outcome) {
return inference.getMarginal(node).getDomainValuePairs().get(outcome);
}
@Override
public Map getMarginals(String node) {
return inference.getMarginal(node).getDomainValuePairs();
}
@Override
public void clearEvidence() {
inference.globalRetraction();
}
@Override
public double[] getMarginalValues(String node) {
return inference.getMarginal(node).getValues();
}
}