naivebayesinference.h

gibbs

#ifndef NAIVEBAYESINFERENCE_H
#define NAIVEBAYESINFERENCE_H

#include "InferenceMethod.h"
#include <list>
#include "VarSet.h"
#include "Distribution.h"
#include "AbstractionSet.h"
#include "JointAbstractionSet.h"

class NaiveBayesInference : public InferenceMethod
{
private:
    AbstractionSet* model;

    vector<GenDistribution*> marginals;
    JointAbstractionSet* jointModel;
    list<int> jointVars;

    void cleanUp() {

        // Free all memory given to us by the inference methods.
        vector<GenDistribution*>::iterator iter;
        for (iter = marginals.begin(); iter != marginals.end(); iter++) {
            delete (*iter);
        }
        marginals.clear();

        delete jointModel;
        jointModel = NULL;
    }

public:
    // Simple constructor
    NaiveBayesInference(AbstractionSet* m)
        : model(m), jointModel(NULL) { /* NOP */ }

    virtual ~NaiveBayesInference() {
        cleanUp();
    }

    virtual void runMarginalInference(const VarSet& evidence) {

        cleanUp();
        marginals = model->createMarginals(evidence);
    }

    virtual void runJointInference(const list<int>& queryVars,
            const VarSet& evidence) {

        cleanUp();
        jointModel = new JointAbstractionSet(model, evidence);
        jointVars = queryVars;
    }

    virtual double singleConditionalLogProb( const list<int>& queryVars,
            const VarSet& evidence, const VarSet& answer) const { 
        return (model->getLogLikelihood(answer) 
                 - model->getLogLikelihood(evidence));
    }

    virtual int getRange(int i) const { return model->getRange(i); }

    //virtual Distribution getMarginal(int i) const { return marginals[i]; };

    virtual double getMarginalProb(int i, double value) const {
        return marginals[i]->getProb(value);
    }

    virtual double getJointProb(const VarConfig& query) const {
        VarSet newquery(query.getNumVars());
        list<int>::const_iterator iter;
        for (iter = jointVars.begin(); iter != jointVars.end(); iter++) {
            newquery[*iter] = query[*iter];
        }

        return jointModel->getProb(newquery);
    }
};

#endif // ndef NAIVEBAYESINFERENCE_H