Sampler.java

Go to the documentation of this file.

package edu.tum.cs.srl.bayesnets.inference;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Vector;
import java.util.regex.Pattern;

import edu.ksu.cis.bnj.ver3.core.BeliefNode;
import edu.tum.cs.bayesnets.inference.SampledDistribution;
import edu.tum.cs.inference.IParameterHandler;
import edu.tum.cs.inference.ParameterHandler;
import edu.tum.cs.srl.bayesnets.bln.AbstractGroundBLN;
import edu.tum.cs.util.Stopwatch;

public abstract class Sampler implements IParameterHandler {
        protected boolean debug = false;
        protected boolean verbose = true;
        protected int numSamples = 1000;
        protected int infoInterval = 100;
        protected ParameterHandler paramHandler;
        protected Vector<Integer> queryVars;
        protected AbstractGroundBLN gbln;
        double samplingTime;
        
        public Sampler(AbstractGroundBLN gbln) throws Exception {
                this.gbln = gbln;
                paramHandler = new ParameterHandler(this);
                paramHandler.add("maxSteps", "setNumSamples");
                paramHandler.add("numSamples", "setNumSamples");
                paramHandler.add("infoInterval", "setInfoInterval");
                paramHandler.add("debug", "setDebugMode");
                paramHandler.add("verbose", "setVerbose");
        }
                
        public Vector<InferenceResult> getResults(SampledDistribution dist) {           
                Vector<InferenceResult> results = new Vector<InferenceResult>();
                for(Integer i : queryVars)
                        results.add(new InferenceResult(dist, i));
                return results;
        }
        
        public void printResults(SampledDistribution dist) {
                ArrayList<InferenceResult> results = new ArrayList<InferenceResult>(getResults(dist));          
                Collections.sort(results);
                for(InferenceResult res : results)
                        res.print();
        }
        
        public double getSamplingTime() {
                return samplingTime;
        }
        
        public void setNumSamples(int n) {
                numSamples = n;
        }
        
        public void setInfoInterval(int n) {
                infoInterval = n;
        }
        
        public SampledDistribution infer() throws Exception {
                Stopwatch sw = new Stopwatch();
                sw.start();
                SampledDistribution ret = _infer();
                samplingTime = sw.getElapsedTimeSecs();
                return ret;
        }
        
        protected abstract SampledDistribution _infer() throws Exception;
        
        public Vector<InferenceResult> inferQueries() throws Exception {
                return getResults(infer());
        }
        
        public String getAlgorithmName() {
                return this.getClass().getSimpleName();
        }
        
        public void setDebugMode(boolean active) {
                debug = active;
        }
        
        public void setVerbose(boolean verbose) {
                this.verbose = verbose;
        }
        
        public ParameterHandler getParameterHandler() {
                return paramHandler;
        }
        
        public void setQueries(Iterable<String> queries) {
                // generate patterns
                Vector<Pattern> patterns = new Vector<Pattern>();
                for(String query : queries) {
                        String p = query;
                        p = Pattern.compile("([,\\(])([a-z][^,\\)]*)").matcher(p).replaceAll("$1.*?");
                        p = p.replace("(", "\\(").replace(")", "\\)") + ".*";                   
                        patterns.add(Pattern.compile(p));
                        //System.out.println("pattern: " + p);
                }
                
                // check all ground variables for matches
                // TODO This should be done more efficiently by explicitly grounding the requested nodes instead of using pattern matchers
                BeliefNode[] nodes = gbln.getGroundNetwork().getNodes();
                queryVars = new Vector<Integer>();
                for(int i = 0; i < nodes.length; i++)
                        for(Pattern pattern : patterns)                         
                                if(pattern.matcher(nodes[i].getName()).matches()) {
                                        queryVars.add(i);
                                        break;
                                }       
        }
}

---

srldb
Author(s): Dominik Jain, Stefan Waldherr, Moritz Tenorth
autogenerated on Fri Jan 11 09:58:30 2013