[doc] Update custom_metric_obj.rst (#8626)

2023-01-06 00:08:25 +03:00 · 2023-01-06 00:08:25 +03:00 · 2b88099c74
commit 2b88099c74
parent e68a152d9e
1 changed files with 1 additions and 0 deletions
--- a/doc/tutorials/custom_metric_obj.rst
+++ b/doc/tutorials/custom_metric_obj.rst
@ -292,6 +292,7 @@ access ``DMatrix``:

    def softprob_obj(labels: np.ndarray, predt: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
        rows = labels.shape[0]
+        classes = predt.shape[1]
        grad = np.zeros((rows, classes), dtype=float)
        hess = np.zeros((rows, classes), dtype=float)
        eps = 1e-6