Spaces:

evaluate-metric
/

roc_auc

Running

Consider replace sklearn.metrics.roc_auc_scorewith torchmetrics.AUROC

by 2catycm - opened Aug 2, 2023

Aug 2, 2023

It is very slow to run the evaluation on CPU, if we have pytorch evalutaion, it should be prefered.
Here is an experiment of the comparison of CPU and GPU roc-auc speed on my dataset with 1 2080Ti GPU.

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment