67 lines
2.0 KiB
C++
67 lines
2.0 KiB
C++
#ifndef HORUS_BAYESNODE_H
|
|
#define HORUS_BAYESNODE_H
|
|
|
|
#include <vector>
|
|
|
|
#include "VarNode.h"
|
|
#include "CptEntry.h"
|
|
#include "Distribution.h"
|
|
#include "Shared.h"
|
|
|
|
using namespace std;
|
|
|
|
|
|
class BayesNode : public VarNode
|
|
{
|
|
public:
|
|
BayesNode (const VarNode& v) : VarNode (v) {}
|
|
BayesNode (VarId, unsigned, int, Distribution*);
|
|
BayesNode (VarId, unsigned, int, const BnNodeSet&, Distribution*);
|
|
|
|
void setParents (const BnNodeSet&);
|
|
void addChild (BayesNode*);
|
|
void setDistribution (Distribution*);
|
|
Distribution* getDistribution (void);
|
|
const ParamSet& getParameters (void);
|
|
ParamSet getRow (int) const;
|
|
void setProbability (int, const CptEntry&, double);
|
|
bool isRoot (void);
|
|
bool isLeaf (void);
|
|
bool hasNeighbors (void) const;
|
|
int getCptSize (void);
|
|
const vector<CptEntry>& getCptEntries (void);
|
|
int getIndexOfParent (const BayesNode*) const;
|
|
string cptEntryToString (const CptEntry&) const;
|
|
string cptEntryToString (int, const CptEntry&) const;
|
|
|
|
const BnNodeSet& getParents (void) const { return parents_; }
|
|
const BnNodeSet& getChilds (void) const { return childs_; }
|
|
|
|
unsigned getRowSize (void) const
|
|
{
|
|
return dist_->params.size() / nrStates();
|
|
}
|
|
|
|
double getProbability (int row, const CptEntry& entry)
|
|
{
|
|
int col = entry.getParameterIndex();
|
|
int idx = (row * getRowSize()) + col;
|
|
return dist_->params[idx];
|
|
}
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN (BayesNode);
|
|
|
|
States getDomainHeaders (void) const;
|
|
friend ostream& operator << (ostream&, const BayesNode&);
|
|
|
|
BnNodeSet parents_;
|
|
BnNodeSet childs_;
|
|
Distribution* dist_;
|
|
};
|
|
|
|
ostream& operator << (ostream&, const BayesNode&);
|
|
|
|
#endif // HORUS_BAYESNODE_H
|
|
|