diff options
Diffstat (limited to 'lm/builder/header_info.hh')
| -rw-r--r-- | lm/builder/header_info.hh | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/lm/builder/header_info.hh b/lm/builder/header_info.hh index 16f3f6090..146195233 100644 --- a/lm/builder/header_info.hh +++ b/lm/builder/header_info.hh @@ -2,16 +2,20 @@ #define LM_BUILDER_HEADER_INFO_H #include <string> +#include <vector> #include <stdint.h> // Some configuration info that is used to add // comments to the beginning of an ARPA file struct HeaderInfo { - const std::string input_file; - const uint64_t token_count; + std::string input_file; + uint64_t token_count; + std::vector<uint64_t> counts_pruned; - HeaderInfo(const std::string& input_file_in, uint64_t token_count_in) - : input_file(input_file_in), token_count(token_count_in) {} + HeaderInfo() {} + + HeaderInfo(const std::string& input_file_in, uint64_t token_count_in, const std::vector<uint64_t> &counts_pruned_in) + : input_file(input_file_in), token_count(token_count_in), counts_pruned(counts_pruned_in) {} // TODO: Add smoothing type // TODO: More info if multiple models were interpolated |