Change docstring of OneHot to avoid misunderstanding

5 years ago · a2bd2c8721
parent 7336ed94fd
commit a2bd2c8721
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/transforms/c_transforms.py
+++ b/mindspore/dataset/transforms/c_transforms.py
@ -32,7 +32,7 @@ class OneHot(cde.OneHotOp):

    Args:
        num_classes (int): Number of classes of the label
-            it should be bigger than or equal to label class number.
+            it should be bigger than largest label number in dataset.

    Raises:
        RuntimeError: feature size is bigger than num_classes.