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

- old
+ new

@@ -26,16 +26,25 @@ DEFINE_TAGGED_SERIALIZER_AFTER_BASE_WITH_VERSION(BaseDocument, 1, 0x00010001, sents, Vs, numGl, numBySentWin, numByWinL, numByWin, numByWinTopicL); template<typename _TopicModel> void update(WeightType* ptr, const _TopicModel& mdl); }; + struct MGLDAArgs : public LDAArgs + { + size_t kL = 1; + size_t t = 3; + std::vector<Float> alphaL = { 0.1 }; + Float alphaMG = 0.1; + Float alphaML = 0.1; + Float etaL = 0.01; + Float gamma = 0.1; + }; + class IMGLDAModel : public ILDAModel { public: using DefaultDocType = DocumentMGLDA<TermWeight::one>; - static IMGLDAModel* create(TermWeight _weight, size_t _KG = 1, size_t _KL = 1, size_t _T = 3, - Float _alphaG = 0.1, Float _alphaL = 0.1, Float _alphaMG = 0.1, Float _alphaML = 0.1, - Float _etaG = 0.01, Float _etaL = 0.01, Float _gamma = 0.1, size_t seed = std::random_device{}(), + static IMGLDAModel* create(TermWeight _weight, const MGLDAArgs& args, bool scalarRng = false); virtual size_t getKL() const = 0; virtual size_t getT() const = 0; virtual Float getAlphaL() const = 0; \ No newline at end of file