[doc] Update custom_metric_obj.rst (#8626)
This commit is contained in:
parent
e68a152d9e
commit
2b88099c74
@ -292,6 +292,7 @@ access ``DMatrix``:
|
|||||||
|
|
||||||
def softprob_obj(labels: np.ndarray, predt: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
|
def softprob_obj(labels: np.ndarray, predt: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
|
||||||
rows = labels.shape[0]
|
rows = labels.shape[0]
|
||||||
|
classes = predt.shape[1]
|
||||||
grad = np.zeros((rows, classes), dtype=float)
|
grad = np.zeros((rows, classes), dtype=float)
|
||||||
hess = np.zeros((rows, classes), dtype=float)
|
hess = np.zeros((rows, classes), dtype=float)
|
||||||
eps = 1e-6
|
eps = 1e-6
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user