parallel_lda_gibbs.h

Go to the documentation of this file.

#ifndef META_PARALLEL_LDA_GIBBS_H_
#define META_PARALLEL_LDA_GIBBS_H_

#include <thread>

#include "parallel/thread_pool.h"
#include "topics/lda_gibbs.h"

namespace meta
{
namespace topics
{

class parallel_lda_gibbs : public lda_gibbs
{
  public:
    using lda_gibbs::lda_gibbs;

    virtual ~parallel_lda_gibbs() = default;

  protected:
    virtual void initialize() override;

    virtual void perform_iteration(uint64_t iter, bool init = false) override;

    virtual void decrease_counts(topic_id topic, term_id term,
                                 doc_id doc) override;

    virtual void increase_counts(topic_id topic, term_id term,
                                 doc_id doc) override;

    virtual double compute_sampling_weight(term_id term, doc_id doc,
                                           topic_id topic) const override;

    parallel::thread_pool pool_;

    std::unordered_map<std::thread::id,
                       std::vector<stats::multinomial<term_id>>> phi_diffs_;
};
}
}

#endif