2014-04-08 20:19:26 +04:00
|
|
|
#include "CountNonTerms.h"
|
|
|
|
#include "moses/Util.h"
|
|
|
|
#include "moses/TargetPhrase.h"
|
|
|
|
#include "moses/StaticData.h"
|
|
|
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
|
|
namespace Moses
|
|
|
|
{
|
|
|
|
CountNonTerms::CountNonTerms(const std::string &line)
|
2015-01-14 14:07:42 +03:00
|
|
|
:StatelessFeatureFunction(line)
|
|
|
|
,m_all(true)
|
|
|
|
,m_sourceSyntax(false)
|
|
|
|
,m_targetSyntax(false)
|
2014-04-08 20:19:26 +04:00
|
|
|
{
|
|
|
|
ReadParameters();
|
|
|
|
}
|
|
|
|
|
2014-07-10 01:35:59 +04:00
|
|
|
void CountNonTerms::EvaluateInIsolation(const Phrase &sourcePhrase
|
2015-01-14 14:07:42 +03:00
|
|
|
, const TargetPhrase &targetPhrase
|
|
|
|
, ScoreComponentCollection &scoreBreakdown
|
|
|
|
, ScoreComponentCollection &estimatedFutureScore) const
|
2014-04-08 20:19:26 +04:00
|
|
|
{
|
|
|
|
const StaticData &staticData = StaticData::Instance();
|
|
|
|
|
|
|
|
vector<float> scores(m_numScoreComponents, 0);
|
|
|
|
size_t indScore = 0;
|
|
|
|
|
|
|
|
if (m_all) {
|
2015-01-14 14:07:42 +03:00
|
|
|
for (size_t i = 0; i < targetPhrase.GetSize(); ++i) {
|
|
|
|
const Word &word = targetPhrase.GetWord(i);
|
|
|
|
if (word.IsNonTerminal()) {
|
|
|
|
++scores[indScore];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
++indScore;
|
2014-04-08 20:19:26 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
if (m_targetSyntax) {
|
2015-01-14 14:07:42 +03:00
|
|
|
for (size_t i = 0; i < targetPhrase.GetSize(); ++i) {
|
|
|
|
const Word &word = targetPhrase.GetWord(i);
|
|
|
|
if (word.IsNonTerminal() && word != staticData.GetOutputDefaultNonTerminal()) {
|
|
|
|
++scores[indScore];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
++indScore;
|
2014-04-08 20:19:26 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
if (m_sourceSyntax) {
|
2015-01-14 14:07:42 +03:00
|
|
|
for (size_t i = 0; i < sourcePhrase.GetSize(); ++i) {
|
|
|
|
const Word &word = sourcePhrase.GetWord(i);
|
|
|
|
if (word.IsNonTerminal() && word != staticData.GetInputDefaultNonTerminal()) {
|
|
|
|
++scores[indScore];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
++indScore;
|
2014-04-08 20:19:26 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
scoreBreakdown.PlusEquals(this, scores);
|
|
|
|
}
|
|
|
|
|
|
|
|
void CountNonTerms::SetParameter(const std::string& key, const std::string& value)
|
|
|
|
{
|
|
|
|
if (key == "all") {
|
|
|
|
m_all = Scan<bool>(value);
|
|
|
|
} else if (key == "source-syntax") {
|
2015-01-14 14:07:42 +03:00
|
|
|
m_sourceSyntax = Scan<bool>(value);
|
2014-04-08 20:19:26 +04:00
|
|
|
} else if (key == "target-syntax") {
|
2015-01-14 14:07:42 +03:00
|
|
|
m_targetSyntax = Scan<bool>(value);
|
2014-04-08 20:19:26 +04:00
|
|
|
} else {
|
|
|
|
StatelessFeatureFunction::SetParameter(key, value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|