File size: 1,928 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
/*
* NBest.h
*
* Created on: 24 Aug 2016
* Author: hieu
*/
#pragma once
#include <queue>
#include <vector>
#include <string>
#include <stdlib.h>
#include "../../Scores.h"
#include "../../ArcLists.h"
namespace Moses2
{
class Scores;
class System;
namespace SCFG
{
class NBest;
class NBests;
class NBestScoreOrderer;
class Manager;
class NBestColl;
class Hypothesis;
/////////////////////////////////////////////////////////////
typedef std::priority_queue<NBest*, std::vector<NBest*>, NBestScoreOrderer> Contenders;
/////////////////////////////////////////////////////////////
class NBest
{
public:
const ArcList *arcList;
size_t arcInd;
typedef std::pair<NBests*, size_t> Child; // key to another NBest
typedef std::vector<Child> Children;
Children children;
NBest(const SCFG::Manager &mgr,
const ArcList &varcList,
size_t vind,
NBestColl &nbestColl);
NBest(const SCFG::Manager &mgr,
const NBest &orig,
size_t childInd,
NBestColl &nbestColl);
void CreateDeviants(
const SCFG::Manager &mgr,
NBestColl &nbestColl,
Contenders &contenders) const;
const Scores &GetScores() const {
return *m_scores;
}
const NBest &GetChild(size_t ind) const;
const std::string &GetString() const {
return m_str;
}
std::string GetStringExclSentenceMarkers() const {
std::string ret = m_str.substr(4, m_str.size() - 10);
return ret;
}
std::string Debug(const System &system) const;
protected:
Scores *m_scores;
std::string m_str;
const SCFG::Hypothesis &GetHypo() const;
void OutputToStream(
const SCFG::Manager &mgr,
std::stringstream &strm) const;
};
/////////////////////////////////////////////////////////////
class NBestScoreOrderer
{
public:
bool operator()(const NBest* a, const NBest* b) const {
return a->GetScores().GetTotalScore() < b->GetScores().GetTotalScore();
}
};
}
}
|