megengine.functional.topk_accuracy

topk_accuracy(logits, target, topk=1)[源代码]

根据给定的预测的logits和真实值标签计算分类准确率。

参数

  • logits (Tensor) – 模型预测值，形为 [batch_size, num_classes] ，表示其属于各类别(class)的概率。

  • target (Tensor) – 真实值标签，int32 类型的一维 Tensor.

  • topk (Union[int, Iterable[int]]) – 指定前 k 个值，该值可以是一个整数，也可以是由整数构成的元组。 默认： 1

返回类型

Union[Tensor, Iterable[Tensor]]

返回

表示分类准确率的张量(一个或多个)，数值介于0.0到1.0之间。

上一页

megengine.functional.topk

下一页

megengine.functional.nn.conv1d