Browse Source

update api document

tags/v0.7.0-beta
anzhengqi 5 years ago
parent
commit
e72e78e2ec
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      mindspore/dataset/transforms/c_transforms.py

+ 4
- 1
mindspore/dataset/transforms/c_transforms.py View File

@@ -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


Loading…
Cancel
Save