Skip to content

Commit

Permalink
Update docs/source/advanced/imbalanced-labels.rst
Browse files Browse the repository at this point in the history
Co-authored-by: Theodore Vasiloudis <theodoros.vasiloudis@gmail.com>
  • Loading branch information
zhjwy9343 and thvasilo authored Feb 18, 2025
1 parent 04bb099 commit 42ff53a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/advanced/imbalanced-labels.rst
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ search for optimal values. Below is an example how to set the `focal loss funcit
gamma: 10.0
alpha: 0.5
Besides the two configurations, you can output the classification results as probabilities of positive and negative
Apart from focal loss and class weights, you can also output the classification results as probabilities of positive and negative
classes by setting the value of ``return_proba`` configuration to be ``true``. By default GraphStorm outputs
classification results using the argmax values, e.g., either 0s or 1s in binary tasks, which equals to using
``0.5`` as the threshold to classify negative from positive samples. With probabilities as outputs, you can use
Expand Down

0 comments on commit 42ff53a

Please sign in to comment.