vendor/tomotopy/src/TopicModel/PLDA.h in tomoto-0.1.4 vs vendor/tomotopy/src/TopicModel/PLDA.h in tomoto-0.2.0

- old
+ new

@@ -1,16 +1,27 @@ #pragma once #include "LLDA.h" namespace tomoto { + struct PLDAArgs : public LDAArgs + { + size_t numLatentTopics = 0; + size_t numTopicsPerLabel = 1; + PLDAArgs setK(size_t _k = 1) const + { + PLDAArgs ret = *this; + ret.k = _k; + return ret; + } + }; + class IPLDAModel : public ILLDAModel { public: using DefaultDocType = DocumentLLDA<TermWeight::one>; - static IPLDAModel* create(TermWeight _weight, size_t _numLatentTopics = 0, size_t _numTopicsPerLabel = 1, - Float alpha = 0.1, Float eta = 0.01, size_t seed = std::random_device{}(), + static IPLDAModel* create(TermWeight _weight, const PLDAArgs& args, bool scalarRng = false); virtual size_t getNumLatentTopics() const = 0; }; } \ No newline at end of file