mosesdecoder/search/config.hh
2015-04-30 12:05:11 +07:00

39 lines
726 B
C++

#ifndef SEARCH_CONFIG__
#define SEARCH_CONFIG__
#include "search/types.hh"
namespace search {
struct NBestConfig {
explicit NBestConfig(unsigned int in_size) {
keep = in_size;
size = in_size;
}
unsigned int keep, size;
};
class Config {
public:
Config(Score lm_weight, unsigned int pop_limit, const NBestConfig &nbest) :
lm_weight_(lm_weight), pop_limit_(pop_limit), nbest_(nbest) {}
Score LMWeight() const { return lm_weight_; }
unsigned int PopLimit() const { return pop_limit_; }
const NBestConfig &GetNBest() const { return nbest_; }
private:
Score lm_weight_;
unsigned int pop_limit_;
NBestConfig nbest_;
};
} // namespace search
#endif // SEARCH_CONFIG__