diff --git a/paddle/fluid/pybind/box_helper_py.cc b/paddle/fluid/pybind/box_helper_py.cc index a797c9c8bddeb..9969cc557276d 100644 --- a/paddle/fluid/pybind/box_helper_py.cc +++ b/paddle/fluid/pybind/box_helper_py.cc @@ -85,6 +85,19 @@ void BindBoxWrapper(py::module* m) { .def("initialize_auc_runner", &framework::BoxWrapper::InitializeAucRunner, py::call_guard()) .def("init_metric", &framework::BoxWrapper::InitMetric, + py::arg("method"), + py::arg("name"), + py::arg("label_varname"), + py::arg("pred_varname"), + py::arg("cmatch_rank_varname"), + py::arg("mask_varname"), + py::arg("metric_phase"), + py::arg("cmatch_rank_group"), + py::arg("ignore_rank"), + py::arg("bucket_size") = 1000000, + py::arg("mode_collect_in_gpu") = false, + py::arg("max_batch_size") = 0, + py::arg("sample_scale_varnam") = "", py::call_guard()) .def("get_metric_msg", &framework::BoxWrapper::GetMetricMsg, py::call_guard())