Class OneHot
Defined in File transforms.h
Inheritance Relationships
Base Type
public mindspore::dataset::TensorTransform
(Class TensorTransform)
Class Documentation
-
class OneHot : public mindspore::dataset::TensorTransform
Convert the labels into OneHot format.
Public Functions
-
explicit OneHot(int32_t num_classes, double smoothing_rate = 0.0)
Constructor.
- 参数
num_classes – [in] number of classes.
smoothing_rate – [in] smoothing rate default(0.0).
样例/* Define operations */ mindspore::MSTensor constant; auto one_hot_op = transforms::OneHot(10); /* dataset is an instance of Dataset object */ dataset = dataset->Map({one_hot_op}, // operations {"column"}); // input columns
-
~OneHot() override = default
Destructor.
-
explicit OneHot(int32_t num_classes, double smoothing_rate = 0.0)