lda_cvb.h

Go to the documentation of this file.

#ifndef META_TOPICS_LDA_CVB_H_
#define META_TOPICS_LDA_CVB_H_

#include "stats/multinomial.h"
#include "topics/lda_model.h"

namespace meta
{
namespace topics
{

class lda_cvb : public lda_model
{
  public:
    lda_cvb(std::shared_ptr<index::forward_index> idx, uint64_t num_topics,
            double alpha, double beta);

    virtual ~lda_cvb() = default;

    void run(uint64_t num_iters, double convergence = 1e-3);

  protected:
    void initialize();

    double perform_iteration(uint64_t iter);

    virtual double
        compute_term_topic_probability(term_id term,
                                       topic_id topic) const override;

    virtual double compute_doc_topic_probability(doc_id doc,
                                                 topic_id topic) const override;

    std::vector<std::vector<stats::multinomial<topic_id>>> gamma_;

    std::vector<stats::multinomial<term_id>> phi_;

    std::vector<stats::multinomial<topic_id>> theta_;
};
}
}
#endif