doc update

perpetual-ml · Oct 8, 2024 · 1950e07 · 1950e07
1 parent 387f5a2
commit 1950e07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/python/perpetual/booster.py b/python-package/python/perpetual/booster.py
@@ -310,7 +310,7 @@ def predict_proba(self, X, parallel: Union[bool, None] = None) -> np.ndarray:
                 Defaults to `None`.
 
         Returns:
-            np.ndarray: Returns a numpy array of the predictions.
+            np.ndarray, shape (n_samples, n_classes): Returns a numpy array of the class probabilities.
         """
         features_, flat_data, rows, cols = transform_input_frame(X, self.cat_mapping)
         self._validate_features(features_)