Couldn't find the retrieval_recall in torcheval

Hi, I’m not sure it is appropriate for this forum, but I have a problem with the archival.
When I tried to use the retrieval_eval from torcheval, I couldn’t run it because it didn’t exist in the package.
https://pytorch.org/torcheval/main/generated/torcheval.metrics.functional.retrieval_recall.html

Does anyone know the proper version of torcheval for it?
My torcheval version is 0.0.7.

retrieval_recall was added in Nov. 2023 in this PR. However, it seems the wheels were not published to PyPI after Aug. 2023 as seen here so you might need to build from source.

1 Like

Thank you for your prompt reply!
Then, can I ask one more question? (Also, if it is not proper for this forum, please let me know.)

Instead of using torcheval, I tried to use torchmetrics for the same function, RetrievalRecall.
https://lightning.ai/docs/torchmetrics/stable/retrieval/recall.html
While I tried some example code, it only returned one tensor by aggregating values (e.g., mean, min, max, median).
How can I get the Recall@K value ‘per’ index?
Thank you for reading this question.