Shortcuts

mmcls.core.precision

mmcls.core.precision(pred, target, average_mode='macro', thrs=0.0)[源代码]

Calculate precision according to the prediction and target.

参数
  • pred (torch.Tensor | np.array) – The model prediction with shape (N, C).

  • target (torch.Tensor | np.array) – The target of each prediction with shape (N, 1) or (N,).

  • average_mode (str) – The type of averaging performed on the result. Options are ‘macro’ and ‘none’. If ‘none’, the scores for each class are returned. If ‘macro’, calculate metrics for each class, and find their unweighted mean. Defaults to ‘macro’.

  • thrs (Number | tuple[Number], optional) – Predictions with scores under the thresholds are considered negative. Default to 0.

返回

Precision.

返回类型

float | np.array | list[float | np.array]

Args

thrs is number

thrs is tuple

average_mode = “macro”

float

list[float]

average_mode = “none”

np.array

list[np.array]

Read the Docs v: latest
Versions
master
latest
1.x
dev-1.x
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.