diff options
Diffstat (limited to 'gi/pyp-topics/src/mpi-pyp-topics.hh')
| -rw-r--r-- | gi/pyp-topics/src/mpi-pyp-topics.hh | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/gi/pyp-topics/src/mpi-pyp-topics.hh b/gi/pyp-topics/src/mpi-pyp-topics.hh index 4a4433e6..d96bc4e5 100644 --- a/gi/pyp-topics/src/mpi-pyp-topics.hh +++ b/gi/pyp-topics/src/mpi-pyp-topics.hh @@ -16,7 +16,7 @@ #include "mpi-pyp.hh" -#include "corpus.hh" +#include "mpi-corpus.hh" class MPIPYPTopics { public: @@ -37,14 +37,14 @@ public: m_am_root = (m_rank == 0); } - void sample_corpus(const Corpus& corpus, int samples, + void sample_corpus(const MPICorpus& corpus, int samples, int freq_cutoff_start=0, int freq_cutoff_end=0, int freq_cutoff_interval=0, int max_contexts_per_document=0); int sample(const DocumentId& doc, const Term& term); - int max(const DocumentId& doc, const Term& term) const; - int max(const DocumentId& doc) const; + std::pair<int,F> max(const DocumentId& doc, const Term& term) const; + std::pair<int,F> max(const DocumentId& doc) const; int max_topic() const; void set_backoff(const std::string& filename) { |