| 113 | }; |
| 114 | |
| 115 | struct DMRModelObject : public LDAModelObject |
| 116 | { |
| 117 | DMRModelObject() = default; |
| 118 | |
| 119 | using _InitArgs = std::tuple<size_t, size_t, size_t, size_t, size_t, PyObject*, float, float, float, PyObject*, PyObject*, PyObject*>; |
| 120 | DMRModelObject(size_t tw, size_t minCnt, size_t minDf, size_t rmTop, |
| 121 | size_t k, PyObject* alpha, float eta, float sigma, float alphaEps, |
| 122 | PyObject* seed, PyObject* corpus, PyObject* transform); |
| 123 | |
| 124 | tomoto::RawDoc::MiscType convertMisc(const tomoto::RawDoc::MiscType& o) const override; |
| 125 | |
| 126 | std::optional<size_t> addDoc(PyObject* words, const std::string& metadata, PyObject* multiMetadata, bool ignoreEmptyWords = true); |
| 127 | py::UniqueCObj<DocumentObject> makeDoc(PyObject* words, const std::string& metadata, PyObject* multiMetadata); |
| 128 | std::vector<float> getTopicPrior(const std::string& metadata, PyObject* multiMetadata, bool raw = false) const; |
| 129 | py::UniqueCObj<VocabObject> getMetadataDict() const; |
| 130 | py::UniqueCObj<VocabObject> getMultiMetadataDict() const; |
| 131 | py::UniqueObj getLambda() const; |
| 132 | py::UniqueObj getLambdaV2() const; |
| 133 | py::UniqueObj getAlpha() const; |
| 134 | }; |
| 135 | |
| 136 | struct GDMRModelObject : public DMRModelObject |
| 137 | { |
nothing calls this directly
no outgoing calls
no test coverage detected