From a2bd2c8721315c215011c105c4f874424130bf18 Mon Sep 17 00:00:00 2001 From: YangLuo Date: Fri, 28 Aug 2020 11:03:45 +0800 Subject: [PATCH] Change docstring of OneHot to avoid misunderstanding --- mindspore/dataset/transforms/c_transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/dataset/transforms/c_transforms.py b/mindspore/dataset/transforms/c_transforms.py index f4b7cde1fa..f3b20e66a6 100644 --- 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.