mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-08-16 23:10:31 +03:00
93 lines
2.3 KiB
C++
93 lines
2.3 KiB
C++
#include <iostream>
|
|
#include "MaxSpanFreeNonTermSource.h"
|
|
#include "moses/StaticData.h"
|
|
#include "moses/Word.h"
|
|
#include "moses/InputPath.h"
|
|
#include "moses/TargetPhrase.h"
|
|
#include "moses/StackVec.h"
|
|
#include "moses/Range.h"
|
|
#include "moses/ChartCellLabel.h"
|
|
#include "moses/FactorCollection.h"
|
|
|
|
using namespace std;
|
|
|
|
namespace Moses
|
|
{
|
|
MaxSpanFreeNonTermSource::MaxSpanFreeNonTermSource(const std::string &line)
|
|
:StatelessFeatureFunction(1, line)
|
|
,m_maxSpan(2)
|
|
,m_glueTargetLHSStr("S")
|
|
,m_glueTargetLHS(true)
|
|
{
|
|
m_tuneable = false;
|
|
ReadParameters();
|
|
|
|
FactorCollection &fc = FactorCollection::Instance();
|
|
const Factor *factor = fc.AddFactor(m_glueTargetLHSStr, true);
|
|
m_glueTargetLHS.SetFactor(0, factor);
|
|
}
|
|
|
|
void MaxSpanFreeNonTermSource::EvaluateInIsolation(const Phrase &source
|
|
, const TargetPhrase &targetPhrase
|
|
, ScoreComponentCollection &scoreBreakdown
|
|
, ScoreComponentCollection &estimatedScores) const
|
|
{
|
|
targetPhrase.SetRuleSource(source);
|
|
}
|
|
|
|
void MaxSpanFreeNonTermSource::EvaluateWithSourceContext(const InputType &input
|
|
, const InputPath &inputPath
|
|
, const TargetPhrase &targetPhrase
|
|
, const StackVec *stackVec
|
|
, ScoreComponentCollection &scoreBreakdown
|
|
, ScoreComponentCollection *estimatedScores) const
|
|
{
|
|
const Word &targetLHS = targetPhrase.GetTargetLHS();
|
|
|
|
if (targetLHS == m_glueTargetLHS) {
|
|
// don't delete glue rules
|
|
return;
|
|
}
|
|
|
|
const Phrase *source = targetPhrase.GetRuleSource();
|
|
assert(source);
|
|
float score = 0;
|
|
|
|
if (source->Front().IsNonTerminal()) {
|
|
const ChartCellLabel &cell = *stackVec->front();
|
|
if (cell.GetCoverage().GetNumWordsCovered() > m_maxSpan) {
|
|
score = - std::numeric_limits<float>::infinity();
|
|
}
|
|
}
|
|
|
|
if (source->Back().IsNonTerminal()) {
|
|
const ChartCellLabel &cell = *stackVec->back();
|
|
if (cell.GetCoverage().GetNumWordsCovered() > m_maxSpan) {
|
|
score = - std::numeric_limits<float>::infinity();
|
|
}
|
|
}
|
|
|
|
|
|
scoreBreakdown.PlusEquals(this, score);
|
|
|
|
}
|
|
|
|
|
|
void MaxSpanFreeNonTermSource::SetParameter(const std::string& key, const std::string& value)
|
|
{
|
|
if (key == "max-span") {
|
|
m_maxSpan = Scan<int>(value);
|
|
} else {
|
|
StatelessFeatureFunction::SetParameter(key, value);
|
|
}
|
|
}
|
|
|
|
std::vector<float> MaxSpanFreeNonTermSource::DefaultWeights() const
|
|
{
|
|
std::vector<float> ret(1, 1);
|
|
return ret;
|
|
}
|
|
|
|
}
|
|
|