CertifaiAI / cdle-traininglabs

Certifai Training Labs Repository using Eclipse Deeplearning4j
https://certifai.ai/
Apache License 2.0
41 stars 70 forks source link

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

Closed KianYang-Lee closed 4 days ago

KianYang-Lee commented 3 years ago

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 commented 3 years ago

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

KianYang-Lee commented 4 days ago

Closing due to inactivity