Method SLDAModelObject

src/python/handler/py_SLDA.cpp:15–104

}

SLDAModelObject::SLDAModelObject(size_t tw, size_t minCnt, size_t minDf, size_t rmTop,
	size_t k, PyObject* vars, PyObject* alpha, float eta,
	PyObject* mu, PyObject* nuSq, PyObject* glmCoef,
	PyObject* seed, PyObject* corpus, PyObject* transform)
{
	tomoto::SLDAArgs margs;
	margs.k = k;
	if (alpha)
	{
		margs.alpha = broadcastObj<tomoto::Float>(alpha, margs.k,
			[&]() { return "`alpha` must be an instance of `float` or `List[float]` with length `k` (given " + py::repr(alpha) + ")"; }
		);
	}
	margs.eta = eta;
	if (seed && seed != Py_None && !py::toCpp<size_t>(seed, margs.seed))
	{
		throw py::ValueError{ "`seed` must be an integer or None." };
	}

	vector<string> varTypeStrs;
	if (vars && vars != Py_None && !py::toCpp<vector<string>>(vars, varTypeStrs))
	{
		throw py::ValueError{ "`vars` must be an iterable of str." };
	}
	for (auto& s : varTypeStrs)
	{
		tomoto::ISLDAModel::GLM t;
		if (s == "l") t = tomoto::ISLDAModel::GLM::linear;
		else if (s == "b") t = tomoto::ISLDAModel::GLM::binary_logistic;
		else throw py::ValueError{ "Unknown var type '" + s + "'" };
		margs.vars.emplace_back(t);
	}

	float fTemp;
	if (mu)
	{
		if (py::toCpp<float>(mu, fTemp))
		{
			margs.mu.resize(varTypeStrs.size(), fTemp);
		}
		else if (py::clearError(), py::toCpp<vector<tomoto::Float>>(mu, margs.mu))
		{

		}
		else
		{
			throw py::ValueError{ "`mu` must be a float or an iterable of float with length same as `vars`." };
		}
	}

	if (nuSq)
	{
		if (py::toCpp<float>(nuSq, fTemp))
		{
			margs.nuSq.resize(varTypeStrs.size(), fTemp);
		}
		else if (py::clearError(), py::toCpp<vector<tomoto::Float>>(nuSq, margs.nuSq))
		{