diff options
Diffstat (limited to 'vowpalwabbit/lda_core.cc')
| -rw-r--r-- | vowpalwabbit/lda_core.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/vowpalwabbit/lda_core.cc b/vowpalwabbit/lda_core.cc index c7be0d9f..db2281a2 100644 --- a/vowpalwabbit/lda_core.cc +++ b/vowpalwabbit/lda_core.cc @@ -495,7 +495,7 @@ v_array<float> old_gamma; for (size_t k =0; k<all.lda; k++)
new_gamma[k] = new_gamma[k]*v[k]+all.lda_alpha;
}
- while (average_diff(all, old_gamma.begin, new_gamma.begin) > 0.001);
+ while (average_diff(all, old_gamma.begin, new_gamma.begin) > all.lda_epsilon);
ec->topic_predictions.erase();
ec->topic_predictions.resize(all.lda);
@@ -748,6 +748,7 @@ learner* setup(vw&all, std::vector<std::string>&opts, po::variables_map& vm) ("lda_alpha", po::value<float>(&all.lda_alpha), "Prior on sparsity of per-document topic weights")
("lda_rho", po::value<float>(&all.lda_rho), "Prior on sparsity of topic distributions")
("lda_D", po::value<float>(&all.lda_D), "Number of documents")
+ ("lda_epsilon", po::value<float>(&all.lda_epsilon), "Loop convergence threshold")
("minibatch", po::value<size_t>(&all.minibatch), "Minibatch size, for LDA");
po::parsed_options parsed = po::command_line_parser(opts).
|