diff options
Diffstat (limited to 'dtrain/ksampler.h')
| -rw-r--r-- | dtrain/ksampler.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/dtrain/ksampler.h b/dtrain/ksampler.h index 0783f98b..f52fb649 100644 --- a/dtrain/ksampler.h +++ b/dtrain/ksampler.h @@ -30,7 +30,7 @@ struct KSampler : public HypSampler vector<ScoredHyp>* GetSamples() { return &s_; } void ScoredSamples(const Hypergraph& forest) { - s_.clear(); + s_.clear(); sz_ = f_count_ = 0; std::vector<HypergraphSampler::Hypothesis> samples; HypergraphSampler::sample_hypotheses(forest, k_, prng_, &samples); for (unsigned i = 0; i < k_; ++i) { @@ -41,6 +41,8 @@ struct KSampler : public HypSampler h.rank = i; h.score = scorer_->Score(h.w, *ref_, i, src_len_); s_.push_back(h); + sz_++; + f_count_ += h.f.size(); } } }; |