NeuralEngine
A Game Engine with embeded Machine Learning algorithms based on Gaussian Processes.
FgDiscreteGraph.h
1
11#pragma once
12
13#include <MachineLearning/FgIGraph.h>
14#include <MachineLearning/FgDiscreteFactorNode.h>
15#include <MachineLearning/FgDiscreteVariableNode.h>
16
17namespace NeuralEngine
18{
19 namespace MachineLearning
20 {
21 class NE_IMPEXP DiscreteGraph : public IGraph<DiscreteFactorNode, DiscreteVariableNode>
22 {
23 public:
24 DiscreteGraph(std::vector<DiscreteFactorNode*> &factorNodes);
26
27 void ComputeMarginal(int numIter, double tolerance = 1e-6f);
28
29 private:
30 friend class boost::serialization::access;
31
32 template<class Archive>
33 void serialize(Archive& ar, unsigned int version)
34 {
35 ar & boost::serialization::base_object<IGraph>(*this);
36 //ar & mNodes & sIncomming & sOutgoing & mMessageBox & sName & iId;
37 }
38 };
39 }
40}