lda_model.h

Go to the documentation of this file.

#ifndef META_TOPICS_LDA_MODEL_H_
#define META_TOPICS_LDA_MODEL_H_

#include "index/forward_index.h"

MAKE_NUMERIC_IDENTIFIER(topic_id, uint64_t)

namespace meta
{
namespace topics
{

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

    virtual ~lda_model() = default;

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

    void save_doc_topic_distributions(const std::string& filename) const;

    void save_topic_term_distributions(const std::string& filename) const;

    void save(const std::string& prefix) const;

  protected:
    lda_model& operator=(const lda_model&) = delete;

    lda_model(const lda_model&) = delete;

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

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

    std::shared_ptr<index::forward_index> idx_;

    size_t num_topics_;

    size_t num_words_;
};
}
}

#endif