diff options
Diffstat (limited to 'gi/pyp-topics/src/train-contexts.cc')
| -rw-r--r-- | gi/pyp-topics/src/train-contexts.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/gi/pyp-topics/src/train-contexts.cc b/gi/pyp-topics/src/train-contexts.cc index 833565cd..02bb7b76 100644 --- a/gi/pyp-topics/src/train-contexts.cc +++ b/gi/pyp-topics/src/train-contexts.cc @@ -44,6 +44,7 @@ int main(int argc, char **argv) ("samples,s", value<int>()->default_value(10), "number of sampling passes through the data") ("backoff-type", value<string>(), "backoff type: none|simple") ("filter-singleton-contexts", "filter singleton contexts") + ("hierarchical-topics", "Use a backoff hierarchical PYP as the P0 for the document topics distribution.") ; store(parse_command_line(argc, argv, cmdline_options), vm); notify(vm); @@ -63,7 +64,7 @@ int main(int argc, char **argv) // seed the random number generator //mt_init_genrand(time(0)); - PYPTopics model(vm["topics"].as<int>()); + PYPTopics model(vm["topics"].as<int>(), vm.count("hierarchical-topics")); // read the data BackoffGenerator* backoff_gen=0; |