Browse Source

!4183 update OneHot parameter document

Merge pull request !4183 from anzhengqi/I1QKPJ-update-document
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d74fbf8695
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