mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-10-27 03:49:57 +03:00
64 lines
1.6 KiB
C++
64 lines
1.6 KiB
C++
#ifndef MERT_HWCM_SCORER_H_
|
|
#define MERT_HWCM_SCORER_H_
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "StatisticsBasedScorer.h"
|
|
#include "moses/FF/InternalTree.h"
|
|
|
|
using Moses::TreePointer;
|
|
using Moses::InternalTree;
|
|
|
|
namespace MosesTuning
|
|
{
|
|
|
|
|
|
class ScoreStats;
|
|
const size_t kHwcmOrder = 4;
|
|
|
|
/**
|
|
* HWCM scoring (Liu and Gildea 2005), but F1 instead of precision.
|
|
*/
|
|
class HwcmScorer: public StatisticsBasedScorer
|
|
{
|
|
public:
|
|
explicit HwcmScorer(const std::string& config = "");
|
|
~HwcmScorer();
|
|
|
|
virtual void setReferenceFiles(const std::vector<std::string>& referenceFiles);
|
|
virtual void prepareStats(std::size_t sid, const std::string& text, ScoreStats& entry);
|
|
|
|
virtual std::size_t NumberOfScores() const {
|
|
return kHwcmOrder*3;
|
|
}
|
|
|
|
virtual float calculateScore(const std::vector<ScoreStatsType>& comps) const;
|
|
|
|
virtual float getReferenceLength(const std::vector<ScoreStatsType>& totals) const {
|
|
return totals[2];
|
|
}
|
|
|
|
//TODO: actually, we use trees which we store in place of alignment. Maybe use something analogous to Phrase Properties to cleanly store trees?
|
|
bool useAlignment() const {
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
|
|
// data extracted from reference files
|
|
std::vector<TreePointer> m_ref_trees;
|
|
std::vector<std::vector<std::map<std::string, int> > > m_ref_hwc;
|
|
std::vector<std::vector<int> > m_ref_lengths;
|
|
|
|
void extractHeadWordChain(TreePointer tree, std::vector<std::string> & history, std::vector<std::map<std::string, int> > & hwc);
|
|
std::string getHead(TreePointer tree);
|
|
|
|
// no copying allowed
|
|
HwcmScorer(const HwcmScorer&);
|
|
HwcmScorer& operator=(const HwcmScorer&);
|
|
};
|
|
|
|
}
|
|
|
|
#endif // MERT_HWCM_SCORER_H_
|