diff options
Diffstat (limited to 'dtrain/kbestget.h')
| -rw-r--r-- | dtrain/kbestget.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/dtrain/kbestget.h b/dtrain/kbestget.h index ae4588c9..cf466fe4 100644 --- a/dtrain/kbestget.h +++ b/dtrain/kbestget.h @@ -1,8 +1,10 @@ #ifndef _DTRAIN_KBESTGET_H_ #define _DTRAIN_KBESTGET_H_ + #include "kbest.h" + namespace dtrain { @@ -65,7 +67,7 @@ struct KBestGetter : public DecoderObserver if (!d) break; kb.sents.push_back( d->yield); kb.feats.push_back( d->feature_values ); - kb.model_scores.push_back( d->score ); + kb.model_scores.push_back( log(d->score) ); } } @@ -83,7 +85,7 @@ struct KBestGetter : public DecoderObserver if (!d) break; kb.sents.push_back( d->yield); kb.feats.push_back( d->feature_values ); - kb.model_scores.push_back( d->score ); + kb.model_scores.push_back( log(d->score) ); } } }; |