mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-29 06:52:34 +03:00
58 lines
1.3 KiB
C++
58 lines
1.3 KiB
C++
#ifndef __SCORER_FACTORY_H
|
|
#define __SCORER_FACTORY_H
|
|
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
#include <iostream>
|
|
#include <iterator>
|
|
#include <set>
|
|
#include <sstream>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "Types.h"
|
|
#include "Scorer.h"
|
|
#include "BleuScorer.h"
|
|
#include "PerScorer.h"
|
|
#include "TerScorer.h"
|
|
#include "CderScorer.h"
|
|
#include "MergeScorer.h"
|
|
|
|
using namespace std;
|
|
|
|
class ScorerFactory
|
|
{
|
|
public:
|
|
static vector<string> getTypes() {
|
|
vector<string> types;
|
|
types.push_back(string("BLEU"));
|
|
types.push_back(string("PER"));
|
|
types.push_back(string("TER"));
|
|
types.push_back(string("CDER"));
|
|
types.push_back(string("MERGE"));
|
|
return types;
|
|
}
|
|
|
|
static Scorer* getScorer(const string& type, const string& config = "") {
|
|
if (type == "BLEU") {
|
|
return (BleuScorer*) new BleuScorer(config);
|
|
} else if (type == "PER") {
|
|
return (PerScorer*) new PerScorer(config);
|
|
} else if (type == "TER") {
|
|
return (TerScorer*) new TerScorer(config);
|
|
} else if (type == "CDER") {
|
|
return (CderScorer*) new CderScorer(config);
|
|
} else if (type == "MERGE") {
|
|
return (MergeScorer*) new MergeScorer(config);
|
|
} else {
|
|
throw runtime_error("Unknown scorer type: " + type);
|
|
}
|
|
}
|
|
|
|
private:
|
|
ScorerFactory() {}
|
|
~ScorerFactory() {}
|
|
};
|
|
|
|
#endif // __SCORER_FACTORY_H
|