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

- old
+ new

@@ -54,15 +54,24 @@ } template<typename _TopicModel> void update(WeightType* ptr, const _TopicModel& mdl); }; + struct HDPArgs : public LDAArgs + { + Float gamma = 0.1; + + HDPArgs() + { + k = 2; + } + }; + class IHDPModel : public ILDAModel { public: using DefaultDocType = DocumentHDP<TermWeight::one>; - static IHDPModel* create(TermWeight _weight, size_t _K = 1, - Float alpha = 0.1, Float eta = 0.01, Float gamma = 0.1, size_t seed = std::random_device{}(), + static IHDPModel* create(TermWeight _weight, const HDPArgs& args, bool scalarRng = false); virtual Float getGamma() const = 0; virtual size_t getTotalTables() const = 0; virtual size_t getLiveK() const = 0; \ No newline at end of file