Shortcuts

mmcls.core.support

mmcls.core.support(pred, target, average_mode='macro')[source]

Calculate the total number of occurrences of each label according to the prediction and target.

Parameters
  • 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 sum. Defaults to ‘macro’.

Returns

Support.

  • If the average_mode is set to macro, the function returns a single float.

  • If the average_mode is set to none, the function returns a np.array with shape C.

Return type

float | np.array

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

Free document hosting provided by Read the Docs.