mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-26 05:14:36 +03:00
105 lines
2.7 KiB
C++
105 lines
2.7 KiB
C++
#include "mmsapt.h"
|
|
#include <boost/foreach.hpp>
|
|
#include <boost/format.hpp>
|
|
#include <boost/tokenizer.hpp>
|
|
#include <boost/shared_ptr.hpp>
|
|
#include <algorithm>
|
|
#include <iostream>
|
|
|
|
using namespace Moses;
|
|
using namespace bitext;
|
|
using namespace std;
|
|
using namespace boost;
|
|
|
|
vector<FactorType> fo(1,FactorType(0));
|
|
|
|
class SimplePhrase : public Moses::Phrase
|
|
{
|
|
vector<FactorType> const m_fo; // factor order
|
|
public:
|
|
SimplePhrase(): m_fo(1,FactorType(0)) {}
|
|
|
|
void init(string const& s)
|
|
{
|
|
istringstream buf(s); string w;
|
|
while (buf >> w)
|
|
{
|
|
Word wrd;
|
|
this->AddWord().CreateFromString(Input,m_fo,StringPiece(w),false,false);
|
|
}
|
|
}
|
|
};
|
|
|
|
class TargetPhraseIndexSorter
|
|
{
|
|
TargetPhraseCollection const& my_tpc;
|
|
CompareTargetPhrase cmp;
|
|
public:
|
|
TargetPhraseIndexSorter(TargetPhraseCollection const& tpc) : my_tpc(tpc) {}
|
|
bool operator()(size_t a, size_t b) const
|
|
{
|
|
return cmp(*my_tpc[a], *my_tpc[b]);
|
|
}
|
|
};
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
Parameter params;
|
|
if (!params.LoadParam(argc,argv) || !StaticData::LoadDataStatic(¶ms, argv[0]))
|
|
exit(1);
|
|
|
|
Mmsapt* PT = NULL;
|
|
BOOST_FOREACH(PhraseDictionary* pd, PhraseDictionary::GetColl())
|
|
if ((PT = dynamic_cast<Mmsapt*>(pd))) break;
|
|
vector<string> const& fname = PT->GetFeatureNames();
|
|
|
|
// vector<FeatureFunction*> const& ffs = FeatureFunction::GetFeatureFunctions();
|
|
|
|
string line;
|
|
while (getline(cin,line))
|
|
{
|
|
SimplePhrase p; p.init(line);
|
|
cout << p << endl;
|
|
TargetPhraseCollection const* trg = PT->GetTargetPhraseCollectionLEGACY(p);
|
|
if (!trg) continue;
|
|
vector<size_t> order(trg->GetSize());
|
|
for (size_t i = 0; i < order.size(); ++i) order[i] = i;
|
|
sort(order.begin(),order.end(),TargetPhraseIndexSorter(*trg));
|
|
size_t k = 0;
|
|
// size_t precision =
|
|
cout.precision(2);
|
|
|
|
BOOST_FOREACH(size_t i, order)
|
|
{
|
|
Phrase const& phr = static_cast<Phrase const&>(*(*trg)[i]);
|
|
cout << setw(3) << ++k << " " << phr << endl;
|
|
ScoreComponentCollection const& scc = (*trg)[i]->GetScoreBreakdown();
|
|
ScoreComponentCollection::IndexPair idx = scc.GetIndexes(PT);
|
|
FVector const& scores = scc.GetScoresVector();
|
|
cout << " ";
|
|
for (size_t k = idx.first; k < idx.second; ++k)
|
|
cout << " " << format("%10.10s") % fname[k-idx.first];
|
|
cout << endl;
|
|
cout << " ";
|
|
for (size_t k = idx.first; k < idx.second; ++k)
|
|
{
|
|
if (fname[k-idx.first].substr(0,3) == "log")
|
|
{
|
|
if(scores[k] < 0)
|
|
cout << " " << format("%10d") % round(exp(-scores[k]));
|
|
else
|
|
cout << " " << format("%10d") % round(exp(scores[k]));
|
|
}
|
|
else
|
|
cout << " " << format("%10.8f") % exp(scores[k]);
|
|
}
|
|
cout << endl;
|
|
}
|
|
PT->Release(trg);
|
|
}
|
|
exit(0);
|
|
}
|
|
|
|
|
|
|