Sha256: b34651a6d52759582d5df6b285c316c7a6a818cb987fd53f0960cf57f77550fa
Contents?: true
Size: 1014 Bytes
Versions: 5
Compression:
Stored size: 1014 Bytes
Contents
#pragma once
#include "LDA.h"
namespace tomoto
{
template<TermWeight _tw>
struct DocumentPT : public DocumentLDA<_tw>
{
using BaseDocument = DocumentLDA<_tw>;
using DocumentLDA<_tw>::DocumentLDA;
using WeightType = typename DocumentLDA<_tw>::WeightType;
uint64_t pseudoDoc = 0;
DEFINE_SERIALIZER_AFTER_BASE_WITH_VERSION(BaseDocument, 0, pseudoDoc);
DEFINE_TAGGED_SERIALIZER_AFTER_BASE_WITH_VERSION(BaseDocument, 1, 0x00010001, pseudoDoc);
};
struct PTArgs : public LDAArgs
{
size_t p = 0;
Float lambda = 0.01;
};
class IPTModel : public ILDAModel
{
public:
using DefaultDocType = DocumentPT<TermWeight::one>;
static IPTModel* create(TermWeight _weight, const PTArgs& args,
bool scalarRng = false);
virtual size_t getP() const = 0;
virtual std::vector<Float> getTopicsFromPseudoDoc(const DocumentBase* doc, bool normalize = true) const = 0;
virtual std::vector<std::pair<Tid, Float>> getTopicsFromPseudoDocSorted(const DocumentBase* doc, size_t topN) const = 0;
};
}
Version data entries
5 entries across 5 versions & 1 rubygems