File size: 1,895 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 |
/*
* KBestExtractor.cpp
*
* Created on: 2 Aug 2016
* Author: hieu
*/
#include <boost/foreach.hpp>
#include <sstream>
#include "KBestExtractor.h"
#include "../Manager.h"
#include "../Hypothesis.h"
#include "../Stacks.h"
#include "../Stack.h"
#include "../Sentence.h"
#include "../../System.h"
#include "../../Scores.h"
#include "../../legacy/Util2.h"
using namespace std;
namespace Moses2
{
//bool g_debug = false;
namespace SCFG
{
/////////////////////////////////////////////////////////////
KBestExtractor::KBestExtractor(const SCFG::Manager &mgr)
:m_mgr(mgr)
{
}
KBestExtractor::~KBestExtractor()
{
}
void KBestExtractor::OutputToStream(std::stringstream &strm)
{
//cerr << "1" << flush;
const Stack &lastStack = m_mgr.GetStacks().GetLastStack();
UTIL_THROW_IF2(lastStack.GetColl().size() != 1, "Only suppose to be 1 hypo coll in last stack");
UTIL_THROW_IF2(lastStack.GetColl().begin()->second == NULL, "NULL hypo collection");
const Hypotheses &hypos = lastStack.GetColl().begin()->second->GetSortedAndPrunedHypos(m_mgr, m_mgr.arcLists);
UTIL_THROW_IF2(hypos.size() != 1, "Only suppose to be 1 hypo in collection");
const HypothesisBase *hypo = hypos[0];
const ArcLists &arcLists = m_mgr.arcLists;
const ArcList &arcList = arcLists.GetArcList(hypo);
NBests &nbests = m_nbestColl.GetOrCreateNBests(m_mgr, arcList);
size_t ind = 0;
while (nbests.Extend(m_mgr, m_nbestColl, ind)) {
const NBest &deriv = nbests.Get(ind);
strm << m_mgr.GetTranslationId() << " ||| ";
//cerr << "1" << flush;
strm << deriv.GetStringExclSentenceMarkers();
//cerr << "2" << flush;
strm << " ||| ";
deriv.GetScores().OutputBreakdown(strm, m_mgr.system);
//cerr << "3" << flush;
strm << "||| ";
strm << deriv.GetScores().GetTotalScore();
//cerr << "4" << flush;
strm << endl;
++ind;
}
}
}
} /* namespace Moses2 */
|