cdle-traininglabs icon indicating copy to clipboard operation
cdle-traininglabs copied to clipboard

Binary Classification examples did not mention about threshold value when using Softmax

Open KianYang-Lee opened this issue 3 years ago • 1 comments

Describe the Enhancement Suggestions For binary classification example, threshold is a hyperparameter to be set to produce different model evaluation metrics. Examples need to be given on how to set threshold for it, else the performance of using softmax and sigmoid would be different.

KianYang-Lee avatar Mar 24 '21 13:03 KianYang-Lee

Refer to this for examples on using Evaluation to modify threshold https://deeplearning4j.org/api/latest/org/nd4j/evaluation/classification/Evaluation.html

KianYang-Lee avatar Mar 24 '21 13:03 KianYang-Lee