47 lines
1.4 KiB
C
47 lines
1.4 KiB
C
|
#ifndef BAYESIAN_NETWORK_H
|
||
|
|
#define BAYESIAN_NETWORK_H
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
class BayesianNode;
|
||
|
|
class Distribution;
|
||
|
|
|
||
|
|
class BayesianNetwork
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// constructs
|
||
|
|
BayesianNetwork (void);
|
||
|
|
// destruct
|
||
|
|
virtual ~BayesianNetwork (void);
|
||
|
|
// methods
|
||
|
|
virtual void addNode (string, vector<BayesianNode*>, int, int);
|
||
|
|
virtual void addNode (string, vector<BayesianNode*>,
|
||
|
|
double*, int, vector<string>);
|
||
|
|
BayesianNode* getNode (string) const;
|
||
|
|
void addDistribution (int, double*, int, vector<string>);
|
||
|
|
vector<BayesianNode*> getNodes (void) const;
|
||
|
|
vector<BayesianNode*> getRootNodes (void) const;
|
||
|
|
vector<BayesianNode*> getLeafNodes (void) const;
|
||
|
|
bool isPolyTree (void) const;
|
||
|
|
void printNetwork (void) const;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
// members
|
||
|
|
vector<BayesianNode*> nodes_;
|
||
|
|
vector<Distribution*> dists_;
|
||
|
|
|
||
|
|
private:
|
||
|
|
BayesianNetwork (const BayesianNetwork&); // disallow copy
|
||
|
|
void operator= (const BayesianNetwork&); // disallow assign
|
||
|
|
bool containsCycle (void) const;
|
||
|
|
bool containsCycle (int, int, vector<bool>&) const;
|
||
|
|
int getIndexOf (const BayesianNode*) const;
|
||
|
|
vector<int> getAdjacentVertexes (int) const ;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // BAYESIAN_NETWORK_H
|
||
|
|
|