From e72e78e2ec4b8cbd671329b250b35344cf817bcb Mon Sep 17 00:00:00 2001 From: anzhengqi Date: Fri, 7 Aug 2020 20:06:50 +0800 Subject: [PATCH] update api document --- mindspore/dataset/transforms/c_transforms.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mindspore/dataset/transforms/c_transforms.py b/mindspore/dataset/transforms/c_transforms.py index 41a31e7c50..0f3c8d145b 100644 --- a/mindspore/dataset/transforms/c_transforms.py +++ b/mindspore/dataset/transforms/c_transforms.py @@ -31,7 +31,10 @@ class OneHot(cde.OneHotOp): Tensor operation to apply one hot encoding. Args: - num_classes (int): Number of classes of the label. + num_classes (int): Number of classes of the label, it should be bigger than feature size. + + Raises: + RuntimeError: feature size is bigger than num_classes. """ @check_num_classes