Evaluation - ROC/AUC/AUPRC improvements

deeplearning4j / deeplearning4j

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

http://deeplearning4j.konduit.ai

Apache License 2.0

13.6k stars 3.83k forks source link

Evaluation - ROC/AUC/AUPRC improvements #3634

Closed AlexDBlack closed 7 years ago

AlexDBlack commented 7 years ago

Use case: Given a ROC instance (and hence an AUC/AUPRC curve), add methods for the following functionality:

get recall (and threshold) at X precision
get precision (and threshold) at X recall
get confusion matrix at specified threshold

AlexDBlack commented 7 years ago

https://github.com/deeplearning4j/deeplearning4j/pull/3636

lock[bot] commented 5 years ago

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.