diff options
Diffstat (limited to 'utils/training/k_mixture_model.h')
-rw-r--r-- | utils/training/k_mixture_model.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/utils/training/k_mixture_model.h b/utils/training/k_mixture_model.h index 2d9816b..6e48796 100644 --- a/utils/training/k_mixture_model.h +++ b/utils/training/k_mixture_model.h @@ -56,8 +56,10 @@ static inline parameter_t compute_B(corpus_count_t N, corpus_count_t n_1){ /* Note: re-check this, to see if we can remove if statement. */ /* Please consider B_2 is no less than 2 in paper. */ +#if 0 if ( 0 == T - n_1 && 0 == N - n_0 - n_1 ) return 2; +#endif parameter_t B = (T - n_1 ) / (parameter_t) (N - n_0 - n_1); return B; @@ -75,7 +77,7 @@ static inline parameter_t compute_Pr_G_3(corpus_count_t k, return alpha * (1 - gamma); if ( k > 1 ) { - return alpha * gamma / (B - 1) * pow((1 - 1 / (B - 1)) , k - 2); + return (alpha * gamma / (B - 1)) * pow((1 - 1 / (B - 1)) , k - 2); } assert(false); |