torcheval.metrics.BinaryRecallAtFixedPrecision

class torcheval.metrics.BinaryRecallAtFixedPrecision(*, min_precision: float, device: device | None = None)

返回在二分类任务中给定最小精度时的最高可能召回值。

它的功能版本是 torcheval.metrics.functional.binary_recall_at_fixed_precision()。

Parameters:

min_precision (float) – 最小精度阈值

示例：

>>> import torch
>>> from torcheval.metrics import BinaryRecallAtFixedPrecision
>>> metric = BinaryRecallAtFixedPrecision(min_precision=0.5)
>>> input = torch.tensor([0.1, 0.4, 0.6, 0.6, 0.6, 0.35, 0.8])
>>> target = torch.tensor([0, 0, 1, 1, 1, 1, 1])
>>> metric.update(input, target)
>>> metric.compute()
(torch.tensor(1.0), torch.tensor(0.35))

__init__(*, min_precision: float, device: device | None = None) → None

初始化一个度量对象及其内部状态。

使用 self._add_state() 来初始化你的度量类的状态变量。 状态变量应该是 torch.Tensor，一个 torch.Tensor 的列表，一个以 torch.Tensor 为值的字典， 或者一个 torch.Tensor 的双端队列。

方法

__init__(*, min_precision[, device])	初始化一个度量对象及其内部状态。
compute()	实现此方法以从状态变量计算并返回最终指标值。
load_state_dict(state_dict[, strict])	从state_dict加载度量状态变量。
merge_state(metrics)	实现此方法以将当前度量的状态变量更新为当前度量和输入度量的合并状态。
reset()	将度量状态变量重置为其默认值。
state_dict()	将度量状态变量保存在state_dict中。
to(device, *args, **kwargs)	将度量状态变量中的张量移动到设备。
update(input, target)	实现此方法以更新您的指标类的状态变量。

属性

device

Metric.to()的最后一个输入设备。