diff options
| author | Chris Dyer <cdyer@cs.cmu.edu> | 2012-04-16 18:20:33 -0400 |
|---|---|---|
| committer | Chris Dyer <cdyer@cs.cmu.edu> | 2012-04-16 18:20:33 -0400 |
| commit | caf1a688db7b446642581c3f69a9aea720735a8f (patch) | |
| tree | 4d550b6d6be2b58d26d3d703d5b09f74d3101484 /utils | |
| parent | b0d1a4f0c9718412691f279b9092f18073cf5094 (diff) | |
rst sampler
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/weights.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/utils/weights.cc b/utils/weights.cc index ac407dfb..39c18474 100644 --- a/utils/weights.cc +++ b/utils/weights.cc @@ -144,8 +144,10 @@ void Weights::ShowLargestFeatures(const vector<weight_t>& w) { vector<int> fnums(w.size()); for (int i = 0; i < w.size(); ++i) fnums[i] = i; + int nf = FD::NumFeats(); + if (nf > 10) nf = 10; vector<int>::iterator mid = fnums.begin(); - mid += (w.size() > 10 ? 10 : w.size()); + mid += nf; partial_sort(fnums.begin(), mid, fnums.end(), FComp(w)); cerr << "TOP FEATURES:"; for (vector<int>::iterator i = fnums.begin(); i != mid; ++i) { |