lda_scvb.h

Go to the documentation of this file.

#ifndef META_TOPICS_LDA_SCVB_H_
#define META_TOPICS_LDA_SCVB_H_

#include "topics/lda_model.h"

namespace meta
{
namespace topics
{

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

    virtual ~lda_scvb() = default;

    virtual void run(uint64_t num_iters, double convergence = 0) override;

  protected:
    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;

  private:
    void initialize(std::mt19937& gen);

    void perform_iteration(uint64_t iter, const std::vector<doc_id>& docs);

    std::vector<std::vector<double>> topic_term_count_;

    std::vector<std::vector<double>> doc_topic_count_;

    std::vector<double> topic_count_;

    const double alpha_;
    const double beta_;
    const uint64_t minibatch_size_;
};
}
}

#endif