57 lines
2.0 KiB
C++
57 lines
2.0 KiB
C++
#ifndef BP_BP_NODE_H
|
|
#define BP_BP_NODE_H
|
|
|
|
#include <vector>
|
|
#include <map>
|
|
#include <deque>
|
|
#include <string>
|
|
|
|
#include "BayesianNode.h"
|
|
|
|
using namespace std;
|
|
|
|
class BpNode : public BayesianNode
|
|
{
|
|
public:
|
|
// constructs
|
|
BpNode (string, vector<BayesianNode*>, Distribution* dist, int = -1);
|
|
// destruct
|
|
~BpNode (void);
|
|
// methods
|
|
static void enableParallelSchedule (void);
|
|
void allocateMemory (void);
|
|
double* getPiValues (void) const;
|
|
double getPiValue (int) const;
|
|
void setPiValue (int, double);
|
|
double* getLambdaValues (void) const;
|
|
double getLambdaValue (int) const;
|
|
void setLambdaValue (int, double);
|
|
double* getPiMessages (BpNode*) const;
|
|
double getPiMessage (BpNode*, int) const;
|
|
void setPiMessage (BpNode*, int, double);
|
|
double* getLambdaMessages (BpNode*) const;
|
|
double getLambdaMessage (BpNode*, int) const;
|
|
void setLambdaMessage (BpNode*, int, double);
|
|
double* getBeliefs (void);
|
|
double getBeliefChange (void);
|
|
void normalizeMessages (void);
|
|
void swapMessages (void);
|
|
|
|
private:
|
|
BpNode (const BpNode&); // disallow copy
|
|
void operator= (const BpNode&); // disallow assign
|
|
// members
|
|
double* lambdaValues_;
|
|
double* piValues_;
|
|
map<BpNode*, double*> piMessages_;
|
|
map<BpNode*, double*> lambdaMessages_;
|
|
map<BpNode*, double*>* newPiMessages_;
|
|
map<BpNode*, double*>* newLambdaMessages_;
|
|
double* oldBeliefs_;
|
|
static bool parallelSchedule_;
|
|
static const double MAX_CHANGE_ = 1.0;
|
|
};
|
|
|
|
#endif // BP_BP_NODE_H
|
|
|