mosesdecoder/mert/Reference.h

112 lines
2.5 KiB
C
Raw Normal View History

#ifndef MERT_REFERENCE_H_
#define MERT_REFERENCE_H_
#include <algorithm>
#include <climits>
#include <vector>
#include "Ngram.h"
namespace MosesTuning
{
2013-05-29 21:16:15 +04:00
/**
* Reference class represents reference translations for an output
* translation used in calculating BLEU score.
*/
2013-05-29 21:16:15 +04:00
class Reference
{
public:
// for m_length
typedef std::vector<std::size_t>::iterator iterator;
typedef std::vector<std::size_t>::const_iterator const_iterator;
Reference() : m_counts(new NgramCounts) { }
2013-05-29 21:16:15 +04:00
~Reference() {
delete m_counts;
}
2013-05-29 21:16:15 +04:00
NgramCounts* get_counts() {
return m_counts;
}
const NgramCounts* get_counts() const {
return m_counts;
}
2013-05-29 21:16:15 +04:00
iterator begin() {
return m_length.begin();
}
const_iterator begin() const {
return m_length.begin();
}
iterator end() {
return m_length.end();
}
const_iterator end() const {
return m_length.end();
}
2013-05-29 21:16:15 +04:00
void push_back(std::size_t len) {
m_length.push_back(len);
}
2013-05-29 21:16:15 +04:00
std::size_t num_references() const {
return m_length.size();
}
int CalcAverage() const;
int CalcClosest(std::size_t length) const;
int CalcShortest() const;
2013-05-29 21:16:15 +04:00
private:
NgramCounts* m_counts;
// multiple reference lengths
std::vector<std::size_t> m_length;
};
2012-04-26 00:44:05 +04:00
// TODO(tetsuok): fix this function and related stuff.
// "average" reference length should not be calculated at sentence-level unlike "closest".
2013-05-29 21:16:15 +04:00
inline int Reference::CalcAverage() const
{
int total = 0;
for (std::size_t i = 0; i < m_length.size(); ++i) {
total += m_length[i];
}
return static_cast<int>(
2013-05-29 21:16:15 +04:00
static_cast<float>(total) / m_length.size());
}
2013-05-29 21:16:15 +04:00
inline int Reference::CalcClosest(std::size_t length) const
{
int min_diff = INT_MAX;
int closest_ref_id = 0; // an index of the closest reference translation
for (std::size_t i = 0; i < m_length.size(); ++i) {
const int ref_length = m_length[i];
const int length_diff = abs(ref_length - static_cast<int>(length));
const int abs_min_diff = abs(min_diff);
// Look for the closest reference
if (length_diff < abs_min_diff) {
min_diff = ref_length - length;
closest_ref_id = i;
// if two references has the same closest length, take the shortest
} else if (length_diff == abs_min_diff) {
if (ref_length < static_cast<int>(m_length[closest_ref_id])) {
closest_ref_id = i;
}
}
}
return static_cast<int>(m_length[closest_ref_id]);
}
2013-05-29 21:16:15 +04:00
inline int Reference::CalcShortest() const
{
return *std::min_element(m_length.begin(), m_length.end());
}
}
#endif // MERT_REFERENCE_H_