73 lines
1.4 KiB
C++
73 lines
1.4 KiB
C++
#ifndef YAP_PACKAGES_CLPBN_HORUS_WEIGHTEDBP_H_
|
|
#define YAP_PACKAGES_CLPBN_HORUS_WEIGHTEDBP_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "BeliefProp.h"
|
|
|
|
|
|
class WeightedLink : public BpLink
|
|
{
|
|
public:
|
|
WeightedLink (FacNode* fn, VarNode* vn, size_t idx, unsigned weight)
|
|
: BpLink (fn, vn), index_(idx), weight_(weight),
|
|
pwdMsg_(vn->range(), LogAware::one()) { }
|
|
|
|
size_t index (void) const { return index_; }
|
|
|
|
unsigned weight (void) const { return weight_; }
|
|
|
|
const Params& powMessage (void) const { return pwdMsg_; }
|
|
|
|
void updateMessage (void);
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN (WeightedLink);
|
|
|
|
size_t index_;
|
|
unsigned weight_;
|
|
Params pwdMsg_;
|
|
};
|
|
|
|
|
|
|
|
inline void
|
|
WeightedLink::updateMessage (void)
|
|
{
|
|
pwdMsg_ = *nextMsg_;
|
|
swap (currMsg_, nextMsg_);
|
|
LogAware::pow (pwdMsg_, weight_);
|
|
}
|
|
|
|
|
|
|
|
class WeightedBp : public BeliefProp
|
|
{
|
|
public:
|
|
WeightedBp (const FactorGraph& fg,
|
|
const std::vector<std::vector<unsigned>>& weights)
|
|
: BeliefProp (fg), weights_(weights) { }
|
|
|
|
~WeightedBp (void);
|
|
|
|
Params getPosterioriOf (VarId);
|
|
|
|
private:
|
|
void createLinks (void);
|
|
|
|
void maxResidualSchedule (void);
|
|
|
|
void calcFactorToVarMsg (BpLink*);
|
|
|
|
Params getVarToFactorMsg (const BpLink*) const;
|
|
|
|
void printLinkInformation (void) const;
|
|
|
|
std::vector<std::vector<unsigned>> weights_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN (WeightedBp);
|
|
};
|
|
|
|
#endif // YAP_PACKAGES_CLPBN_HORUS_WEIGHTEDBP_H_
|
|
|