fl_sim.models.top_n_accuracy

fl_sim.models.top_n_accuracy(preds: Tensor, labels: Tensor, n: int = 1) → float

Top-n accuracy.

参数:

• preds (torch.Tensor) -- Shape (batch_size, n_classes) or (batch_size, n_classes, d_1, ..., d_n).

• labels (torch.Tensor) -- Shape (batch_size,) or (batch_size, d_1, ..., d_n).

返回:

The top-n accuracy.

返回类型:

float