torcheval.metrics.functional.frequency_at_k

torcheval.metrics.functional.frequency_at_k(input: Tensor, k: float) → Tensor

根据频率列表和阈值 k 计算频率。生成一个二进制列表，指示频率是否小于 k。

参数:

• input (Tensor) – 预测的未归一化的分数（通常称为 logits）。

• k (float) – 频率阈值。k 不应为负值。

示例

>>> import torch
>>> from torcheval.metrics.functional import frequency
>>> input = torch.tensor([0.3, 0.1, 0.6])
>>> frequency(input, k=0.5)
tensor([1.0000, 1.0000, 0.0000])