Browse Source

!5406 Change docstring of OneHot to avoid misunderstanding

Merge pull request !5406 from luoyang/c-api
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c006e7b901
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/dataset/transforms/c_transforms.py

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

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


Loading…
Cancel
Save