From 0298806fd0dc05139b0b731ba2bc5f5582a109ee Mon Sep 17 00:00:00 2001 From: Aref Razavi Date: Mon, 27 Mar 2023 21:32:14 -0600 Subject: [PATCH] Fix available options for mode in the docstring. --- tensorflow_addons/metrics/hamming.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_addons/metrics/hamming.py b/tensorflow_addons/metrics/hamming.py index 215bf9ce7a..a1e13e987f 100644 --- a/tensorflow_addons/metrics/hamming.py +++ b/tensorflow_addons/metrics/hamming.py @@ -72,7 +72,7 @@ def hamming_loss_fn( threshold: Elements of `y_pred` greater than threshold are converted to be 1, and the rest 0. If threshold is None, the argmax is converted to 1, and the rest 0. - mode: multi-class or multi-label. + mode: 'multiclass' or 'multilabel'. Returns: hamming loss: float.