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