From 5871b2c2f0778922bcfff69df286bd0532080790 Mon Sep 17 00:00:00 2001 From: Sergii Dymchenko Date: Tue, 17 Sep 2024 22:23:05 -0700 Subject: [PATCH] Fix misspelled docstring argument (#208) Summary: Pull Request resolved: https://github.com/pytorch/torcheval/pull/208 Reviewed By: JKSenthil Differential Revision: D62916001 Pulled By: kit1980 fbshipit-source-id: 646659405afb1565d344a9b261ac1c4293da61f4 --- .../functional/classification/binary_normalized_entropy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torcheval/metrics/functional/classification/binary_normalized_entropy.py b/torcheval/metrics/functional/classification/binary_normalized_entropy.py index c968f0c1..bfd23ad1 100644 --- a/torcheval/metrics/functional/classification/binary_normalized_entropy.py +++ b/torcheval/metrics/functional/classification/binary_normalized_entropy.py @@ -33,7 +33,7 @@ def binary_normalized_entropy( weight (Tensor): Optional. A manual rescaling weight to match input tensor shape (num_tasks, num_samples). num_tasks (int): Number of tasks that need BinaryNormalizedEntropy calculation. Default value is 1. - from_logit (bool): A boolean indicator whether the predicted value `y_pred` is + from_logits (bool): A boolean indicator whether the predicted value `y_pred` is a floating-point logit value (i.e., value in [-inf, inf] when `from_logits=True`) or a probablity value (i.e., value in [0., 1.] when `from_logits=False`) Default value is False.