Method addDoc

src/python/handler/py_GDMR.cpp:78–128

}

std::optional<size_t> GDMRModelObject::addDoc(PyObject* words, PyObject* numericMetadata, const std::string& metadata, PyObject* multiMetadata, bool ignoreEmptyWords)
{
	if (isPrepared) throw py::RuntimeError{ "cannot add_doc() after train()" };
	auto* inst = getInst<tomoto::IGDMRModel>();
	if (PyUnicode_Check(words))
	{
		if (PyErr_WarnEx(PyExc_RuntimeWarning, "`words` should be an iterable of str.", 1)) throw py::ExcPropagation{};
	}

	tomoto::RawDoc raw = buildRawDoc(words);
	raw.misc["metadata"] = metadata;
	if (multiMetadata)
	{
		vector<string> multiMetadataValue;
		if (!py::toCpp<vector<string>>(multiMetadata, multiMetadataValue))
		{
			throw py::ValueError{
				"`multi_metadata` must be an instance of `List[str]` (but given " + py::repr(multiMetadata) + ")"
			};
		}
		raw.misc["multi_metadata"] = move(multiMetadataValue);
	}
	vector<tomoto::Float> nmd;
	if (!py::toCpp(numericMetadata, nmd)) 
	{
		throw py::ValueError{
			"`numeric_metadata` must be an iterable of float (but given " + py::repr(numericMetadata) + ")"
		};
	}
	for (auto x : nmd)
	{
		if (!isfinite(x)) throw py::ValueError{ "`numeric_metadata` has non-finite value (" + py::reprFromCpp(nmd) + ")." };
	}
	raw.misc["numeric_metadata"] = move(nmd);
	try
	{
		auto ret = inst->addDoc(raw);
		return py::buildPyValue(ret);
	}
	catch (const tomoto::exc::EmptyWordArgument&)
	{
		if (ignoreEmptyWords)
		{
			return std::nullopt;
		}
		else
		{
			throw;
		}
	}
}

py::UniqueCObj<DocumentObject> GDMRModelObject::makeDoc(PyObject* words, PyObject* numericMetadata, const std::string& metadata, PyObject* multiMetadata)
{