Sha256: eafd32ca4462c0870858e20a38214037a0a7bc960575b35907540b47cbf5ae22
Contents?: true
Size: 1.61 KB
Versions: 9
Compression:
Stored size: 1.61 KB
Contents
#pragma once
#include "LDA.h"
namespace tomoto
{
template<TermWeight _tw>
struct DocumentPA : public DocumentLDA<_tw>
{
using BaseDocument = DocumentLDA<_tw>;
using DocumentLDA<_tw>::DocumentLDA;
using WeightType = typename DocumentLDA<_tw>::WeightType;
tvector<Tid> Z2s;
Eigen::Matrix<WeightType, -1, -1> numByTopic1_2;
template<typename _TopicModel> void update(WeightType* ptr, const _TopicModel& mdl);
DEFINE_SERIALIZER_AFTER_BASE_WITH_VERSION(BaseDocument, 0, Z2s);
DEFINE_TAGGED_SERIALIZER_AFTER_BASE_WITH_VERSION(BaseDocument, 1, 0x00010001, Z2s);
};
struct PAArgs : public LDAArgs
{
size_t k2 = 1;
std::vector<Float> subalpha = { 0.1 };
};
class IPAModel : public ILDAModel
{
public:
using DefaultDocType = DocumentPA<TermWeight::one>;
static IPAModel* create(TermWeight _weight, const PAArgs& args,
bool scalarRng = false);
virtual size_t getDirichletEstIteration() const = 0;
virtual void setDirichletEstIteration(size_t iter) = 0;
virtual size_t getK2() const = 0;
virtual Float getSubAlpha(Tid k1, Tid k2) const = 0;
virtual std::vector<Float> getSubAlpha(Tid k1) const = 0;
virtual std::vector<Float> getSubTopicBySuperTopic(Tid k, bool normalize = true) const = 0;
virtual std::vector<std::pair<Tid, Float>> getSubTopicBySuperTopicSorted(Tid k, size_t topN) const = 0;
virtual std::vector<Float> getSubTopicsByDoc(const DocumentBase* doc, bool normalize = true) const = 0;
virtual std::vector<std::pair<Tid, Float>> getSubTopicsByDocSorted(const DocumentBase* doc, size_t topN) const = 0;
virtual std::vector<uint64_t> getCountBySuperTopic() const = 0;
};
}
Version data entries
9 entries across 9 versions & 1 rubygems