Package ai.djl.modality.cv.transform
Class OneHot
- java.lang.Object
-
- ai.djl.modality.cv.transform.OneHot
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description NDArray
transform(NDArray array)
Applies theTransform
to the givenNDArray
.
-
-
-
Constructor Detail
-
OneHot
public OneHot(int numClass)
Creates aOneHot
Transform
that converts the sparse label to one-hot label.- Parameters:
numClass
- number of classes
-
-