Function bind_sequence_segmenter

tools/python/src/sequence_segmenter.cpp:777–823

// ----------------------------------------------------------------------------------------

void bind_sequence_segmenter(py::module& m)
{
    py::class_<segmenter_params>(m, "segmenter_params",
"This class is used to define all the optional parameters to the    \n\
train_sequence_segmenter() and cross_validate_sequence_segmenter() routines.   ")
        .def(py::init<>())
        .def_readwrite("use_BIO_model", &segmenter_params::use_BIO_model)
        .def_readwrite("use_high_order_features", &segmenter_params::use_high_order_features)
        .def_readwrite("allow_negative_weights", &segmenter_params::allow_negative_weights)
        .def_readwrite("window_size", &segmenter_params::window_size)
        .def_readwrite("num_threads", &segmenter_params::num_threads)
        .def_readwrite("epsilon", &segmenter_params::epsilon)
        .def_readwrite("max_cache_size", &segmenter_params::max_cache_size)
        .def_readwrite("C", &segmenter_params::C, "SVM C parameter")
        .def_readwrite("be_verbose", &segmenter_params::be_verbose)
        .def("__repr__",&segmenter_params__repr__)
        .def("__str__",&segmenter_params__str__)
        .def(py::pickle(&getstate<segmenter_params>, &setstate<segmenter_params>));

    py::class_<segmenter_type> (m, "segmenter_type", "This object represents a sequence segmenter and is the type of object "
        "returned by the dlib.train_sequence_segmenter() routine.")
        .def("__call__", &segmenter_type::segment_sequence_dense)
        .def("__call__", &segmenter_type::segment_sequence_sparse)
        .def_property_readonly("weights", &segmenter_type::get_weights)
        .def(py::pickle(&getstate<segmenter_type>, &setstate<segmenter_type>));

    py::class_<segmenter_test> (m, "segmenter_test", "This object is the output of the dlib.test_sequence_segmenter() and "
        "dlib.cross_validate_sequence_segmenter() routines.")
        .def_readwrite("precision", &segmenter_test::precision)
        .def_readwrite("recall", &segmenter_test::recall)
        .def_readwrite("f1", &segmenter_test::f1)
        .def("__repr__",&segmenter_test__repr__)
        .def("__str__",&segmenter_test__str__)
        .def(py::pickle(&getstate<segmenter_test>, &setstate<segmenter_test>));

    m.def("train_sequence_segmenter", train_dense, py::arg("samples"), py::arg("segments"), py::arg("params")=segmenter_params());
    m.def("train_sequence_segmenter", train_sparse, py::arg("samples"), py::arg("segments"), py::arg("params")=segmenter_params());


    m.def("test_sequence_segmenter", test_sequence_segmenter1);
    m.def("test_sequence_segmenter", test_sequence_segmenter2);

    m.def("cross_validate_sequence_segmenter", cross_validate_sequence_segmenter1,
        py::arg("samples"), py::arg("segments"), py::arg("folds"), py::arg("params")=segmenter_params());
    m.def("cross_validate_sequence_segmenter", cross_validate_sequence_segmenter2,
        py::arg("samples"), py::arg("segments"), py::arg("folds"), py::arg("params")=segmenter_params());
}