Skip to content

Commit

Permalink
Update metrics.py
Browse files Browse the repository at this point in the history
the doc was editted based on the discussion in the issue:
INT32 Failed on paddle.metric.accuracy: #42845
  • Loading branch information
onecatcn authored and jzhang533 committed May 23, 2022
1 parent 2300d45 commit d5b6eec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/paddle/metric/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -771,7 +771,7 @@ def accuracy(input, label, k=1, correct=None, total=None, name=None):
Args:
input(Tensor): The input of accuracy layer, which is the predictions of network. A Tensor with type float32,float64.
The shape is ``[sample_number, class_dim]`` .
label(Tensor): The label of dataset. Tensor with type int32,int64. The shape is ``[sample_number, 1]`` .
label(Tensor): The label of dataset. Tensor with type int64. The shape is ``[sample_number, 1]`` .
k(int, optional): The top k predictions for each class will be checked. Data type is int64 or int32.
correct(Tensor, optional): The correct predictions count. A Tensor with type int64 or int32.
total(Tensor, optional): The total entries count. A tensor with type int64 or int32.
Expand Down

0 comments on commit d5b6eec

Please sign in to comment.