diff options
author | graehl@gmail.com <graehl@gmail.com@ec762483-ff6d-05da-a07a-a48fb63a330f> | 2010-08-02 07:57:23 +0000 |
---|---|---|
committer | graehl@gmail.com <graehl@gmail.com@ec762483-ff6d-05da-a07a-a48fb63a330f> | 2010-08-02 07:57:23 +0000 |
commit | f9859ad4116733e145d7b8eb31c3cc9318ff7564 (patch) | |
tree | 92f6942fc7fd7066eb400bce6d2cbd2fee46c801 /decoder/hg.cc | |
parent | 6da285dfa7b0a1929dcec882d7e48a585e878d18 (diff) |
fake tdict names for non-ids, push viterbi cost to root in hg, store as feature. type erased fsa feature via virtual interface. made lexical_cast assume C locale for speed.
git-svn-id: https://ws10smt.googlecode.com/svn/trunk@465 ec762483-ff6d-05da-a07a-a48fb63a330f
Diffstat (limited to 'decoder/hg.cc')
-rw-r--r-- | decoder/hg.cc | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/decoder/hg.cc b/decoder/hg.cc index 88e95337..f0238fe3 100644 --- a/decoder/hg.cc +++ b/decoder/hg.cc @@ -61,6 +61,7 @@ struct TropicalValue { } TropicalValue(unsigned v) : v_(v) {} TropicalValue(const prob_t& v) : v_(v) {} +// operator prob_t() const { return v_; } inline TropicalValue& operator+=(const TropicalValue& o) { if (v_ < o.v_) v_ = o.v_; return *this; @@ -90,6 +91,7 @@ struct ViterbiTransitionEventWeightFunction { }; +//TODO: both Compute* methods build sparse vectors with size = whole subhypergraph, for every node. there's no need for that. prob_t Hypergraph::ComputeEdgePosteriors(double scale, vector<prob_t>* posts) const { const ScaledEdgeProb weight(scale); const ScaledTransitionEventWeightFunction w2(scale); @@ -131,6 +133,27 @@ void Hypergraph::PushWeightsToSource(double scale) { } } +namespace { +struct vpusher : public vector<TropicalValue> { + int fid; + vpusher(int fid=0) : fid(fid) { } + void operator()(int n,int /*ei*/,Hypergraph::Edge &e) const { + Hypergraph::TailNodeVector const& t=e.tail_nodes_; + prob_t p=e.edge_prob_; + for (int i=0;i<t.size();++i) + p*=(*this)[t[i]].v_; + e.feature_values_.set_value(fid,log(e.edge_prob_=p/(*this)[n].v_)); + } +}; +} + +void Hypergraph::PushViterbiWeightsToGoal(int fid) { + vpusher vi(fid); + Inside(*this,&vi,ViterbiWeightFunction()); + visit_edges(vi); +} + + void Hypergraph::PushWeightsToGoal(double scale) { vector<prob_t> posts; ComputeEdgePosteriors(scale, &posts); @@ -425,6 +448,7 @@ struct IdCompare { //TODO: if you had parallel arrays associating data w/ each node or edge, you'd want access to reloc_node and reloc_edge - expose in stateful object? void Hypergraph::TopologicallySortNodesAndEdges(int goal_index, const vector<bool>* prune_edges) { + edges_topo_=true; // figure out which nodes are reachable from the goal vector<int> reloc_node(nodes_.size(), -1); vector<int> reloc_edge(edges_.size(), -1); |