mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-10-27 11:59:52 +03:00
71 lines
1.7 KiB
C
71 lines
1.7 KiB
C
|
// -*- c++ -*-
|
||
|
|
// written by Ulrich Germann
|
||
|
|
#pragma once
|
||
|
|
#include "moses/TranslationModel/UG/mm/ug_bitext.h"
|
||
|
|
#include "util/exception.hh"
|
||
|
|
#include "boost/format.hpp"
|
||
|
|
#include "boost/foreach.hpp"
|
||
|
|
|
||
|
|
namespace Moses {
|
||
|
|
namespace bitext
|
||
|
|
{
|
||
|
|
template<typename Token>
|
||
|
|
class
|
||
|
|
PScorePfwd : public PhraseScorer<Token>
|
||
|
|
{
|
||
|
|
float conf;
|
||
|
|
string denom;
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
PScorePfwd(float const c, string d)
|
||
|
|
{
|
||
|
|
this->m_index = -1;
|
||
|
|
conf = c;
|
||
|
|
denom = d;
|
||
|
|
size_t checksum = d.size();
|
||
|
|
BOOST_FOREACH(char const& x, denom)
|
||
|
|
{
|
||
|
|
if (x == '+') { --checksum; continue; }
|
||
|
|
if (x != 'g' && x != 's' && x != 'r') continue;
|
||
|
|
string s = (format("pfwd-%c%.3f") % x % c).str();
|
||
|
|
this->m_feature_names.push_back(s);
|
||
|
|
}
|
||
|
|
this->m_num_feats = this->m_feature_names.size();
|
||
|
|
UTIL_THROW_IF2(this->m_feature_names.size() != checksum,
|
||
|
|
"Unknown parameter in specification '"
|
||
|
|
<< d << "' for Pfwd phrase scorer at " << HERE);
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
operator()(Bitext<Token> const& bt, PhrasePair<Token> & pp,
|
||
|
|
vector<float> * dest = NULL) const
|
||
|
|
{
|
||
|
|
if (!dest) dest = &pp.fvals;
|
||
|
|
if (pp.joint > pp.good1)
|
||
|
|
{
|
||
|
|
pp.joint = pp.good1;
|
||
|
|
// cerr<<bt.toString(pp.p1,0)<<" ::: "<<bt.toString(pp.p2,1)<<endl;
|
||
|
|
// cerr<<pp.joint<<"/"<<pp.good1<<"/"<<pp.raw2<<endl;
|
||
|
|
}
|
||
|
|
size_t i = this->m_index;
|
||
|
|
BOOST_FOREACH(char const& c, this->denom)
|
||
|
|
{
|
||
|
|
switch (c)
|
||
|
|
{
|
||
|
|
case 'g':
|
||
|
|
(*dest)[i++] = log(lbop(pp.good1, pp.joint, conf));
|
||
|
|
break;
|
||
|
|
case 's':
|
||
|
|
(*dest)[i++] = log(lbop(pp.sample1, pp.joint, conf));
|
||
|
|
break;
|
||
|
|
case 'r':
|
||
|
|
(*dest)[i++] = log(lbop(pp.raw1, pp.joint, conf));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|