Method DTModelObject

src/python/handler/py_DT.cpp:15–43

}

DTModelObject::DTModelObject(size_t tw, size_t minCnt, size_t minDf, size_t rmTop,
	size_t k, size_t t, float alphaVar, float etaVar, float phiVar,
	float lrA, float lrB, float lrC,
	PyObject* seed, PyObject* corpus, PyObject* transform)
{
	tomoto::DTArgs margs;
	margs.k = k;
	margs.t = t;
	margs.alpha[0] = alphaVar;
	margs.eta = etaVar;
	margs.phi = phiVar;
	margs.shapeA = lrA;
	margs.shapeB = lrB;
	margs.shapeC = lrC;

	if (seed && seed != Py_None && !py::toCpp<size_t>(seed, margs.seed))
	{
		throw py::ValueError{ "`seed` must be an integer or None." };
	}

	inst = tomoto::IDTModel::create((tomoto::TermWeight)tw, margs);
	if (!inst) throw py::RuntimeError{ "unknown `tw` value" };
	isPrepared = false;
	seedGiven = !!seed;
	minWordCnt = minCnt;
	minWordDf = minDf;
	removeTopWord = rmTop;
	insertCorpus(corpus, transform);
}

std::optional<size_t> DTModelObject::addDoc(PyObject* words, size_t timepoint, bool ignoreEmptyWords)
{