diff options
Diffstat (limited to 'klm/lm/config.hh')
| -rw-r--r-- | klm/lm/config.hh | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/klm/lm/config.hh b/klm/lm/config.hh index 1f7762be..6c7fe39b 100644 --- a/klm/lm/config.hh +++ b/klm/lm/config.hh @@ -3,6 +3,7 @@ #include <iosfwd> +#include "lm/lm_exception.hh" #include "util/mmap.hh" /* Configuration for ngram model. Separate header to reduce pollution. */ @@ -27,13 +28,16 @@ struct Config { // ONLY EFFECTIVE WHEN READING ARPA - typedef enum {THROW_UP, COMPLAIN, SILENT} WarningAction; // What to do when <unk> isn't in the provided model. WarningAction unknown_missing; // What to do when <s> or </s> is missing from the model. // If THROW_UP, the exception will be of type util::SpecialWordMissingException. WarningAction sentence_marker_missing; + // What to do with a positive log probability. For COMPLAIN and SILENT, map + // to 0. + WarningAction positive_log_probability; + // The probability to substitute for <unk> if it's missing from the model. // No effect if the model has <unk> or unknown_missing == THROW_UP. float unknown_missing_logprob; |