pytorch-lightningのtorchmetricsにある Recallの定義
torchmetrics.Recallを使いたかったけど、説明に陽性陰性の定義方法が書いてなかった。ソース辿っていくと
metrics/torchmetrics/functional/classification/stat_scores.py
にその定義が書いてあった。
具体的には
https://github.com/PyTorchLightning/metrics/blob/03a8520e4c6d0282260820a03d86a9f260314694/torchmetrics/functional/classification/stat_scores.py#L65
に
pos_pred, neg_pred = preds == 1, preds == 0
とあったので、1が陽性で0が陰性である。 ドキュメントに書いて欲しい。バイナリで指定してでだけではどっちか不安になってしまう。しかし、提供されているだけありがたい。自分で書くとこういうところは面倒だ。