mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-11-14 19:08:50 +03:00
leak
This commit is contained in:
parent
d58bbf158d
commit
74b635e42f
144
contrib/other-builds/moses2/Distortion.cpp
Normal file
144
contrib/other-builds/moses2/Distortion.cpp
Normal file
@ -0,0 +1,144 @@
|
||||
/*
|
||||
* Distortion.cpp
|
||||
*
|
||||
* Created on: 28 Oct 2015
|
||||
* Author: hieu
|
||||
*/
|
||||
|
||||
#include "Distortion.h"
|
||||
#include "Hypothesis.h"
|
||||
#include "Manager.h"
|
||||
#include "moses/Range.h"
|
||||
#include "moses/Bitmap.h"
|
||||
|
||||
struct DistortionState_traditional : public Moses::FFState {
|
||||
Moses::Range range;
|
||||
int first_gap;
|
||||
DistortionState_traditional(const Moses::Range& wr, int fg) : range(wr), first_gap(fg) {}
|
||||
|
||||
size_t hash() const {
|
||||
return range.GetEndPos();
|
||||
}
|
||||
virtual bool operator==(const FFState& other) const {
|
||||
const DistortionState_traditional& o =
|
||||
static_cast<const DistortionState_traditional&>(other);
|
||||
return range.GetEndPos() == o.range.GetEndPos();
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
|
||||
///////////////////////////////////////////////////////////////////////
|
||||
Distortion::Distortion(size_t startInd, const std::string &line)
|
||||
:StatefulFeatureFunction(startInd, line)
|
||||
{
|
||||
ReadParameters();
|
||||
}
|
||||
|
||||
Distortion::~Distortion() {
|
||||
// TODO Auto-generated destructor stub
|
||||
}
|
||||
|
||||
const Moses::FFState* Distortion::EmptyHypothesisState(const Manager &mgr, const Phrase &input) const
|
||||
{
|
||||
MemPool &pool = mgr.GetPool();
|
||||
|
||||
// fake previous translated phrase start and end
|
||||
size_t start = NOT_FOUND;
|
||||
size_t end = NOT_FOUND;
|
||||
/*
|
||||
if (input.m_frontSpanCoveredLength > 0) {
|
||||
// can happen with --continue-partial-translation
|
||||
start = 0;
|
||||
end = input.m_frontSpanCoveredLength -1;
|
||||
}
|
||||
*/
|
||||
return new (pool.Allocate<DistortionState_traditional>()) DistortionState_traditional(
|
||||
Moses::Range(start, end),
|
||||
NOT_FOUND);
|
||||
|
||||
}
|
||||
|
||||
void
|
||||
Distortion::EvaluateInIsolation(const System &system,
|
||||
const PhraseBase &source, const TargetPhrase &targetPhrase,
|
||||
Scores &scores,
|
||||
Scores *estimatedFutureScores) const
|
||||
{
|
||||
}
|
||||
|
||||
Moses::FFState* Distortion::EvaluateWhenApplied(const Manager &mgr,
|
||||
const Hypothesis &hypo,
|
||||
const Moses::FFState &prevState,
|
||||
Scores &score) const
|
||||
{
|
||||
MemPool &pool = mgr.GetPool();
|
||||
|
||||
const DistortionState_traditional &prev = static_cast<const DistortionState_traditional&>(prevState);
|
||||
SCORE distortionScore = CalculateDistortionScore(
|
||||
prev.range,
|
||||
hypo.GetRange(),
|
||||
prev.first_gap);
|
||||
|
||||
score.PlusEquals(mgr.GetSystem(), *this, distortionScore);
|
||||
|
||||
DistortionState_traditional* res = new (pool.Allocate<DistortionState_traditional>()) DistortionState_traditional(
|
||||
hypo.GetRange(),
|
||||
hypo.GetBitmap().GetFirstGapPos());
|
||||
return res;
|
||||
|
||||
}
|
||||
|
||||
SCORE Distortion::CalculateDistortionScore(const Moses::Range &prev, const Moses::Range &curr, const int FirstGap) const
|
||||
{
|
||||
bool useEarlyDistortionCost = false;
|
||||
if(!useEarlyDistortionCost) {
|
||||
return - (SCORE) ComputeDistortionDistance(prev, curr);
|
||||
} else {
|
||||
/* Pay distortion score as soon as possible, from Moore and Quirk MT Summit 2007
|
||||
Definitions:
|
||||
S : current source range
|
||||
S' : last translated source phrase range
|
||||
S'' : longest fully-translated initial segment
|
||||
*/
|
||||
|
||||
int prefixEndPos = (int)FirstGap-1;
|
||||
if((int)FirstGap==-1)
|
||||
prefixEndPos = -1;
|
||||
|
||||
// case1: S is adjacent to S'' => return 0
|
||||
if ((int) curr.GetStartPos() == prefixEndPos+1) {
|
||||
//IFVERBOSE(4) std::cerr<< "MQ07disto:case1" << std::endl;
|
||||
return 0;
|
||||
}
|
||||
|
||||
// case2: S is to the left of S' => return 2(length(S))
|
||||
if ((int) curr.GetEndPos() < (int) prev.GetEndPos()) {
|
||||
//IFVERBOSE(4) std::cerr<< "MQ07disto:case2" << std::endl;
|
||||
return (float) -2*(int)curr.GetNumWordsCovered();
|
||||
}
|
||||
|
||||
// case3: S' is a subsequence of S'' => return 2(nbWordBetween(S,S'')+length(S))
|
||||
if ((int) prev.GetEndPos() <= prefixEndPos) {
|
||||
//IFVERBOSE(4) std::cerr<< "MQ07disto:case3" << std::endl;
|
||||
int z = (int)curr.GetStartPos()-prefixEndPos - 1;
|
||||
return (float) -2*(z + (int)curr.GetNumWordsCovered());
|
||||
}
|
||||
|
||||
// case4: otherwise => return 2(nbWordBetween(S,S')+length(S))
|
||||
//IFVERBOSE(4) std::cerr<< "MQ07disto:case4" << std::endl;
|
||||
return (float) -2*((int)curr.GetNumWordsBetween(prev) + (int)curr.GetNumWordsCovered());
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
int Distortion::ComputeDistortionDistance(const Moses::Range& prev, const Moses::Range& current) const
|
||||
{
|
||||
int dist = 0;
|
||||
if (prev.GetNumWordsCovered() == 0) {
|
||||
dist = current.GetStartPos();
|
||||
} else {
|
||||
dist = (int)prev.GetEndPos() - (int)current.GetStartPos() + 1 ;
|
||||
}
|
||||
return abs(dist);
|
||||
}
|
42
contrib/other-builds/moses2/Distortion.h
Normal file
42
contrib/other-builds/moses2/Distortion.h
Normal file
@ -0,0 +1,42 @@
|
||||
/*
|
||||
* Distortion.h
|
||||
*
|
||||
* Created on: 28 Oct 2015
|
||||
* Author: hieu
|
||||
*/
|
||||
|
||||
#ifndef DISTORTION_H_
|
||||
#define DISTORTION_H_
|
||||
|
||||
#include "StatefulFeatureFunction.h"
|
||||
#include "moses/Range.h"
|
||||
#include "TypeDef.h"
|
||||
|
||||
class Distortion : public StatefulFeatureFunction
|
||||
{
|
||||
public:
|
||||
Distortion(size_t startInd, const std::string &line);
|
||||
virtual ~Distortion();
|
||||
|
||||
virtual const Moses::FFState* EmptyHypothesisState(const Manager &mgr, const Phrase &input) const;
|
||||
|
||||
virtual void
|
||||
EvaluateInIsolation(const System &system,
|
||||
const PhraseBase &source,
|
||||
const TargetPhrase &targetPhrase,
|
||||
Scores &scores,
|
||||
Scores *estimatedFutureScores) const;
|
||||
|
||||
virtual Moses::FFState* EvaluateWhenApplied(const Manager &mgr,
|
||||
const Hypothesis &hypo,
|
||||
const Moses::FFState &prevState,
|
||||
Scores &score) const;
|
||||
|
||||
protected:
|
||||
SCORE CalculateDistortionScore(const Moses::Range &prev, const Moses::Range &curr, const int FirstGap) const;
|
||||
|
||||
int ComputeDistortionDistance(const Moses::Range& prev, const Moses::Range& current) const;
|
||||
|
||||
};
|
||||
|
||||
#endif /* DISTORTION_H_ */
|
Loading…
Reference in New Issue
Block a user