Method CoherenceObject

src/python/handler/py_coherence.cpp:7–35

using namespace std;

CoherenceObject::CoherenceObject(PyObject* corpus,
	ProbEstimation pe, Segmentation seg, ConfirmMeasure cm, IndirectMeasure im,
	size_t windowSize, double eps, double gamma, PyObject* targets)
	: model{ pe, windowSize }
{
	this->corpus = py::checkType<CorpusObject>(py::UniqueObj{ corpus }, "`corpus` must be an instance of `tomotopy.utils.Corpus`.");
	this->corpus.incref();

	vector<tomoto::Vid> targetIds;
	py::foreach<string>(targets, [&](const string& w)
	{
		auto wid = this->corpus->getVocabDict().toWid(w);
		if (wid != tomoto::non_vocab_id) targetIds.emplace_back(wid);
	}, "`targets` must be an iterable of `str`.");

	this->model.insertTargets(targetIds.begin(), targetIds.end());

	for (size_t i = 0; i < this->corpus->len(); ++i)
	{
		auto* doc = this->corpus->getDoc(i);
		this->model.insertDoc(
			wordBegin(doc, this->corpus->isIndependent()),
			wordEnd(doc, this->corpus->isIndependent())
		);
	}

	this->seg = seg;
	this->cm = tomoto::coherence::AnyConfirmMeasurer::getInstance(cm, im, targetIds.begin(), targetIds.end(), eps, gamma);
}

double CoherenceObject::getScore(PyObject* words) const
{