afb81b8278
Some checks failed
Docker. / Ubuntu (push) Has been cancelled
User-agent updater. / User-agent (push) Failing after 15s
Lock Threads / lock (push) Failing after 10s
Waiting for answer. / waiting-for-answer (push) Failing after 22s
Needs user action. / needs-user-action (push) Failing after 8s
Can't reproduce. / cant-reproduce (push) Failing after 8s
Close stale issues and PRs / stale (push) Has been cancelled
44 lines
1.9 KiB
C++
44 lines
1.9 KiB
C++
#include <pybind11/pybind11.h>
|
|
#include <pybind11/pytypes.h>
|
|
#include <pybind11/stl.h>
|
|
|
|
#include "../src/nnet_language_identifier.h"
|
|
|
|
namespace pybind11 {
|
|
|
|
using chrome_lang_id::NNetLanguageIdentifier;
|
|
|
|
// This is conventional.
|
|
namespace py = pybind11;
|
|
|
|
PYBIND11_MODULE(pybind_ext, py_module) {
|
|
py::class_<NNetLanguageIdentifier>(py_module, "NNetLanguageIdentifier")
|
|
.def(py::init<const int, const int>(), py::arg("min_num_bytes"),
|
|
py::arg("max_num_bytes"))
|
|
.def("FindLanguage", &NNetLanguageIdentifier::FindLanguage,
|
|
py::arg("text"))
|
|
.def("FindTopNMostFreqLangs",
|
|
&NNetLanguageIdentifier::FindTopNMostFreqLangs, py::arg("text"),
|
|
py::arg("num_langs"))
|
|
.def_readonly_static("kUnknown", &NNetLanguageIdentifier::kUnknown)
|
|
.def_readonly_static("kMinNumBytesToConsider",
|
|
&NNetLanguageIdentifier::kMinNumBytesToConsider)
|
|
.def_readonly_static("kMaxNumBytesToConsider",
|
|
&NNetLanguageIdentifier::kMaxNumBytesToConsider)
|
|
.def_readonly_static("kMaxNumInputBytesToConsider",
|
|
&NNetLanguageIdentifier::kMaxNumInputBytesToConsider)
|
|
.def_readonly_static("kReliabilityThreshold",
|
|
&NNetLanguageIdentifier::kReliabilityThreshold)
|
|
.def_readonly_static("kReliabilityHrBsThreshold",
|
|
&NNetLanguageIdentifier::kReliabilityHrBsThreshold);
|
|
|
|
py::class_<NNetLanguageIdentifier::Result>(py_module, "Result")
|
|
.def_readwrite("language", &NNetLanguageIdentifier::Result::language)
|
|
.def_readwrite("probability",
|
|
&NNetLanguageIdentifier::Result::probability)
|
|
.def_readwrite("is_reliable",
|
|
&NNetLanguageIdentifier::Result::is_reliable)
|
|
.def_readwrite("proportion", &NNetLanguageIdentifier::Result::proportion);
|
|
}
|
|
} // namespace pybind11
|