mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-27 05:55:02 +03:00
247 lines
6.6 KiB
C++
247 lines
6.6 KiB
C++
#include <iostream>
|
|
|
|
#include "ForestRescore.h"
|
|
|
|
#define BOOST_TEST_MODULE MertForestRescore
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
|
|
|
|
using namespace std;
|
|
using namespace MosesTuning;
|
|
|
|
BOOST_AUTO_TEST_CASE(viterbi_simple_lattice)
|
|
{
|
|
Vocab vocab;
|
|
WordVec words;
|
|
string wordStrings[] =
|
|
{"<s>", "</s>", "a", "b", "c", "d", "e", "f", "g"};
|
|
for (size_t i = 0; i < 9; ++i) {
|
|
words.push_back(&(vocab.FindOrAdd((wordStrings[i]))));
|
|
}
|
|
|
|
const string f1 = "foo";
|
|
const string f2 = "bar";
|
|
Graph graph(vocab);
|
|
graph.SetCounts(5,5);
|
|
|
|
Edge* e0 = graph.NewEdge();
|
|
e0->AddWord(words[0]);
|
|
e0->AddFeature(f1, 2.0);
|
|
|
|
Vertex* v0 = graph.NewVertex();
|
|
v0->AddEdge(e0);
|
|
|
|
Edge* e1 = graph.NewEdge();
|
|
e1->AddWord(NULL);
|
|
e1->AddChild(0);
|
|
e1->AddWord(words[2]);
|
|
e1->AddWord(words[3]);
|
|
e1->AddFeature(f1, 1.0);
|
|
e1->AddFeature(f2, 3.0);
|
|
|
|
Vertex* v1 = graph.NewVertex();
|
|
v1->AddEdge(e1);
|
|
|
|
Edge* e2 = graph.NewEdge();
|
|
e2->AddWord(NULL);
|
|
e2->AddChild(1);
|
|
e2->AddWord(words[4]);
|
|
e2->AddWord(words[5]);
|
|
e2->AddFeature(f2, 2.5);
|
|
|
|
Vertex* v2 = graph.NewVertex();
|
|
v2->AddEdge(e2);
|
|
|
|
Edge* e3 = graph.NewEdge();
|
|
e3->AddWord(NULL);
|
|
e3->AddChild(2);
|
|
e3->AddWord(words[6]);
|
|
e3->AddWord(words[7]);
|
|
e3->AddWord(words[8]);
|
|
e3->AddFeature(f1, -1);
|
|
|
|
Vertex* v3 = graph.NewVertex();
|
|
v3->AddEdge(e3);
|
|
|
|
Edge* e4 = graph.NewEdge();
|
|
e4->AddWord(NULL);
|
|
e4->AddChild(3);
|
|
e4->AddWord(words[1]);
|
|
e3->AddFeature(f2, 0.5);
|
|
|
|
Vertex* v4 = graph.NewVertex();
|
|
v4->AddEdge(e4);
|
|
|
|
ReferenceSet references;
|
|
references.AddLine(0, "a b c k e f o", vocab);
|
|
HgHypothesis modelHypo;
|
|
vector<FeatureStatsType> bg(kBleuNgramOrder*2+1);
|
|
SparseVector weights;
|
|
weights.set(f1,2);
|
|
weights.set(f2,1);
|
|
Viterbi(graph, weights, 0, references, 0, bg, &modelHypo);
|
|
BOOST_CHECK_CLOSE(2.0,modelHypo.featureVector.get(f1), 0.0001);
|
|
BOOST_CHECK_CLOSE(6.0,modelHypo.featureVector.get(f2), 0.0001);
|
|
|
|
BOOST_CHECK_EQUAL(words[0]->first, modelHypo.text[0]->first);
|
|
BOOST_CHECK_EQUAL(words[2]->first, modelHypo.text[1]->first);
|
|
BOOST_CHECK_EQUAL(words[3]->first, modelHypo.text[2]->first);
|
|
BOOST_CHECK_EQUAL(words[4]->first, modelHypo.text[3]->first);
|
|
BOOST_CHECK_EQUAL(words[5]->first, modelHypo.text[4]->first);
|
|
BOOST_CHECK_EQUAL(words[6]->first, modelHypo.text[5]->first);
|
|
BOOST_CHECK_EQUAL(words[7]->first, modelHypo.text[6]->first);
|
|
BOOST_CHECK_EQUAL(words[8]->first, modelHypo.text[7]->first);
|
|
BOOST_CHECK_EQUAL(words[1]->first, modelHypo.text[8]->first);
|
|
}
|
|
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(viterbi_3branch_lattice)
|
|
{
|
|
Vocab vocab;
|
|
WordVec words;
|
|
string wordStrings[] =
|
|
{"<s>", "</s>", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k"};
|
|
for (size_t i = 0; i < 13; ++i) {
|
|
words.push_back(&(vocab.FindOrAdd((wordStrings[i]))));
|
|
}
|
|
|
|
const string f1 = "foo";
|
|
const string f2 = "bar";
|
|
Graph graph(vocab);
|
|
graph.SetCounts(5,8);
|
|
|
|
Edge* e0 = graph.NewEdge();
|
|
e0->AddWord(words[0]);
|
|
|
|
Vertex* v0 = graph.NewVertex();
|
|
v0->AddEdge(e0);
|
|
|
|
Edge* e1 = graph.NewEdge();
|
|
e1->AddWord(NULL);
|
|
e1->AddChild(0);
|
|
e1->AddWord(words[2]);
|
|
e1->AddWord(words[3]);
|
|
e1->AddFeature(f1,1);
|
|
e1->AddFeature(f2,1);
|
|
Edge* e5 = graph.NewEdge();
|
|
e5->AddWord(NULL);
|
|
e5->AddChild(0);
|
|
e5->AddWord(words[9]);
|
|
e5->AddWord(words[10]);
|
|
e5->AddFeature(f1,2);
|
|
e5->AddFeature(f2,-2);
|
|
|
|
Vertex* v1 = graph.NewVertex();
|
|
v1->AddEdge(e1);
|
|
v1->AddEdge(e5);
|
|
v1->SetSourceCovered(1);
|
|
|
|
Edge* e2 = graph.NewEdge();
|
|
e2->AddWord(NULL);
|
|
e2->AddChild(1);
|
|
e2->AddWord(words[4]);
|
|
e2->AddWord(words[5]);
|
|
e2->AddFeature(f2,3);
|
|
|
|
Vertex* v2 = graph.NewVertex();
|
|
v2->AddEdge(e2);
|
|
v2->SetSourceCovered(3);
|
|
|
|
Edge* e3 = graph.NewEdge();
|
|
e3->AddWord(NULL);
|
|
e3->AddChild(2);
|
|
e3->AddWord(words[6]);
|
|
e3->AddWord(words[7]);
|
|
e3->AddWord(words[8]);
|
|
e3->AddFeature(f1,1);
|
|
Edge* e6 = graph.NewEdge();
|
|
e6->AddWord(NULL);
|
|
e6->AddChild(2);
|
|
e6->AddWord(words[9]);
|
|
e6->AddWord(words[12]);
|
|
e6->AddFeature(f2,1);
|
|
Edge* e7 = graph.NewEdge();
|
|
e7->AddWord(NULL);
|
|
e7->AddChild(1);
|
|
e7->AddWord(words[11]);
|
|
e7->AddWord(words[12]);
|
|
e7->AddFeature(f1,2);
|
|
e7->AddFeature(f2,3);
|
|
|
|
Vertex* v3 = graph.NewVertex();
|
|
v3->AddEdge(e3);
|
|
v3->AddEdge(e6);
|
|
v3->AddEdge(e7);
|
|
v3->SetSourceCovered(5);
|
|
|
|
Edge* e4 = graph.NewEdge();
|
|
e4->AddWord(NULL);
|
|
e4->AddChild(3);
|
|
e4->AddWord(words[1]);
|
|
|
|
Vertex* v4 = graph.NewVertex();
|
|
v4->AddEdge(e4);
|
|
v4->SetSourceCovered(6);
|
|
|
|
/*Paths || foo || bar || s(2,1)
|
|
ab cd hk || 1 || 5 || 7
|
|
hi cd hk || 2 || 2 || 6
|
|
ab jk || 3 || 4 || 10
|
|
hi jk || 4 || 1 || 9
|
|
ab cd efg || 2 || 4 || 8
|
|
hi cd efg || 3 || 1 || 7
|
|
*/
|
|
|
|
ReferenceSet references;
|
|
references.AddLine(0, "a b c d h k", vocab);
|
|
HgHypothesis modelHypo;
|
|
vector<FeatureStatsType> bg(kBleuNgramOrder*2+1, 0.1);
|
|
SparseVector weights;
|
|
weights.set(f1,2);
|
|
weights.set(f2,1);
|
|
Viterbi(graph, weights, 0, references, 0, bg, &modelHypo);
|
|
BOOST_CHECK_CLOSE(3.0,modelHypo.featureVector.get(f1), 0.0001);
|
|
BOOST_CHECK_CLOSE(4.0,modelHypo.featureVector.get(f2), 0.0001);
|
|
|
|
BOOST_CHECK_EQUAL(6, modelHypo.text.size());
|
|
|
|
//expect ab jk
|
|
BOOST_CHECK_EQUAL(words[0]->first, modelHypo.text[0]->first);
|
|
BOOST_CHECK_EQUAL(words[2]->first, modelHypo.text[1]->first);
|
|
BOOST_CHECK_EQUAL(words[3]->first, modelHypo.text[2]->first);
|
|
BOOST_CHECK_EQUAL(words[11]->first, modelHypo.text[3]->first);
|
|
BOOST_CHECK_EQUAL(words[12]->first, modelHypo.text[4]->first);
|
|
BOOST_CHECK_EQUAL(words[1]->first, modelHypo.text[5]->first);
|
|
|
|
|
|
HgHypothesis hopeHypo;
|
|
Viterbi(graph, weights, 1, references, 0, bg, &hopeHypo);
|
|
//expect abcdhk
|
|
BOOST_CHECK_EQUAL(8, hopeHypo.text.size());
|
|
|
|
BOOST_CHECK_EQUAL(words[0]->first, hopeHypo.text[0]->first);
|
|
BOOST_CHECK_EQUAL(words[2]->first, hopeHypo.text[1]->first);
|
|
BOOST_CHECK_EQUAL(words[3]->first, hopeHypo.text[2]->first);
|
|
BOOST_CHECK_EQUAL(words[4]->first, hopeHypo.text[3]->first);
|
|
BOOST_CHECK_EQUAL(words[5]->first, hopeHypo.text[4]->first);
|
|
BOOST_CHECK_EQUAL(words[9]->first, hopeHypo.text[5]->first);
|
|
BOOST_CHECK_EQUAL(words[12]->first, hopeHypo.text[6]->first);
|
|
BOOST_CHECK_EQUAL(words[1]->first, hopeHypo.text[7]->first);
|
|
|
|
BOOST_CHECK_EQUAL(kBleuNgramOrder*2+1, hopeHypo.bleuStats.size());
|
|
BOOST_CHECK_EQUAL(6, hopeHypo.bleuStats[0]);
|
|
BOOST_CHECK_EQUAL(6, hopeHypo.bleuStats[1]);
|
|
BOOST_CHECK_EQUAL(5, hopeHypo.bleuStats[2]);
|
|
BOOST_CHECK_EQUAL(5, hopeHypo.bleuStats[3]);
|
|
BOOST_CHECK_EQUAL(4, hopeHypo.bleuStats[4]);
|
|
BOOST_CHECK_EQUAL(4, hopeHypo.bleuStats[5]);
|
|
BOOST_CHECK_EQUAL(3, hopeHypo.bleuStats[6]);
|
|
BOOST_CHECK_EQUAL(3, hopeHypo.bleuStats[7]);
|
|
BOOST_CHECK_EQUAL(6, hopeHypo.bleuStats[8]);
|
|
}
|
|
|
|
|
|
|