| 309 | }; |
| 310 | |
| 311 | struct DTModelObject : public LDAModelObject |
| 312 | { |
| 313 | DTModelObject() = default; |
| 314 | |
| 315 | using _InitArgs = std::tuple<size_t, size_t, size_t, size_t, size_t, size_t, float, float, float, float, float, float, PyObject*, PyObject*, PyObject*>; |
| 316 | DTModelObject(size_t tw, size_t minCnt, size_t minDf, size_t rmTop, |
| 317 | size_t k, size_t t, float alphaVar, float etaVar, float phiVar, |
| 318 | float lrA, float lrB, float lrC, |
| 319 | PyObject* seed, PyObject* corpus, PyObject* transform); |
| 320 | |
| 321 | tomoto::RawDoc::MiscType convertMisc(const tomoto::RawDoc::MiscType& o) const override; |
| 322 | std::optional<size_t> addDoc(PyObject* words, size_t timepoint, bool ignoreEmptyWords = true); |
| 323 | py::UniqueCObj<DocumentObject> makeDoc(PyObject* words, size_t timepoint); |
| 324 | std::vector<float> getAlpha2(size_t timepoint) const; |
| 325 | std::vector<float> getPhi(size_t timepoint, size_t topicId) const; |
| 326 | std::vector<std::pair<std::string, tomoto::Float>> getTopicWords(size_t topicId, size_t timepoint, size_t topN = 10) const; |
| 327 | std::vector<float> getTopicWordDist(size_t topicId, size_t timepoint, bool normalize = true) const; |
| 328 | py::UniqueObj getCountByTopic() const; |
| 329 | py::UniqueObj getAlpha() const; |
| 330 | }; |
| 331 | |
| 332 | struct PTModelObject : public LDAModelObject |
| 333 | { |
nothing calls this directly
no outgoing calls
no test coverage detected