Browse Source

[MNT] add a warning to predict_proba of BasicNN

pull/6/head
Gao Enhao 1 year ago
parent
commit
d612987c84
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      ablkit/learning/basic_nn.py

+ 6
- 0
ablkit/learning/basic_nn.py View File

@@ -337,6 +337,12 @@ class BasicNN:
X : List[Any], optional
The input data. Defaults to None.

Warning
-------
This method calculates the probability by applying a softmax function to the output
of the neural network. If your neural network already includes a softmax function
as its final activation, applying softmax again here will lead to incorrect probabilities.

Returns
-------
numpy.ndarray


Loading…
Cancel
Save