forked from sibyl-team/sib
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mf.cpp
68 lines (54 loc) · 1.03 KB
/
mf.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
// This file is part of sibilla : inference in epidemics with Belief Propagation
// Author: Alfredo Braunstein
// Author: Alessandro Ingrosso
// Author: Anna Paola Muntoni
// Author: Indaco Biazzo
#include <string.h>
#include <iostream>
#include <fstream>
#include <sstream>
#include <algorithm>
#include <functional>
#include <algorithm>
#include <assert.h>
#include <tuple>
#include <exception>
#include "mf.h"
#include "cavity.h"
using namespace std;
template<>
char const * MFGraph::name()
{
return "MFGraph";
}
template<>
char const * MFNode::name()
{
return "MFNode";
}
std::ostream & operator<<(std::ostream &o, MFMes const & msg)
{
for (size_t i = 0; i < msg.size(); ++i)
o << msg << " ";
return o << std::endl;
}
MFMes & operator++(MFMes & msg)
{
msg.push_back(0.0);
return msg;
}
MFMes & operator--(MFMes & msg)
{
msg.erase(msg.begin(), msg.begin() + 1);
return msg;
}
template<>
real_t MFGraph::update(int i, real_t damping, bool learn)
{
return 0.0;
}
template<>
real_t MFGraph::loglikelihood() const
{
return 0.0;
}