43 #ifndef __NGRAM_MODEL_INTERNAL_H__
44 #define __NGRAM_MODEL_INTERNAL_H__
98 #define NGRAM_MAX_ORDER 5
100 #define NGRAM_HASH_SIZE 128
102 #define NGRAM_BASEWID(wid) ((wid)&0xffffff)
103 #define NGRAM_CLASSID(wid) (((wid)>>24) & 0x7f)
104 #define NGRAM_CLASSWID(wid,classid) (((classid)<<24) | 0x80000000 | (wid))
105 #define NGRAM_IS_CLASSWID(wid) ((wid)&0x80000000)
107 #define UG_ALLOC_STEP 10
124 int32 * history, int32 n_hist, int32 * n_used);
131 int32 * history, int32 n_hist, int32 * n_used);
166 logmath_t * lmath, int32 n, int32 n_unigram);
172 const char *classdef_file);
183 int32 start_wid,
glist_t classwords);
void(* free)(ngram_model_t *model)
Implementation-specific function for freeing an ngram_model_t.
int32(* score)(ngram_model_t *model, int32 wid, int32 *history, int32 n_hist, int32 *n_used)
Implementation-specific function for querying language model score.
int32 log_zero
Zero probability, cached here for quick lookup.
uint8 n_classes
Number of classes (maximum 128)
uint32 * n_counts
Counts for 1, 2, 3, ...
uint8 n
This is an n-gram model (1, 2, 3, ...).
logmath_t * lmath
Log-math object.
One class definition from a classdef file.
int32 prob1
Probability for this word.
int32 n_hash
Number of buckets in nword_hash (power of 2)
int32 wid
Word ID of this bucket.
float32 lw
Language model scaling factor.
A node in a generic list.
int32 n_words
Number of actual word strings (NOT the same as the number of unigrams, due to class words).
struct ngram_class_s ** classes
Word class definitions.
int32(* raw_score)(ngram_model_t *model, int32 wid, int32 *history, int32 n_hist, int32 *n_used)
Implementation-specific function for querying raw language model probability.
struct ngram_funcs_s * funcs
Implementation-specific methods.
int32 n_words
Number of base words for this class.
int32 tag_wid
Base word ID for this class tag.
int32 log_wip
Log of word insertion penalty.
int32 next
Index of next bucket (or -1 for no collision)
int32 start_wid
Starting base word ID for this class' words.
Custom hash table for additional words.
int32 * tmp_wids
Temporary array of word IDs for ngram_model_get_ngram()
int32 n_hash_inuse
Number of words in nword_hash.
hash_table_t * wid
Mapping of unigram names to word IDs.
Implementation-specific functions for operating on ngram_model_t objects.
uint8 writable
Are word strings writable?
Implementation of ngram_class_t.
char ** word_str
Unigram names.
int32 * prob1
Probability table for base words.
int refcount
Reference count.
uint8 flags
Any other flags we might care about (FIXME: Merge this and writable)
int32 n_1g_alloc
Number of allocated word strings (for new word addition)
int(* apply_weights)(ngram_model_t *model, float32 lw, float32 wip)
Implementation-specific function for applying language model weights.
int32(* add_ug)(ngram_model_t *model, int32 wid, int32 lweight)
Implementation-specific function for adding unigrams.
void(* flush)(ngram_model_t *model)
Implementation-specific function for purging N-Gram cache.
Hash table implementation.
Common implementation of ngram_model_t.