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 NDArraytransform(NDArray array)Applies theTransformto the givenNDArray.
-
-
-
Constructor Detail
-
OneHot
public OneHot(int numClass)
Creates aOneHotTransformthat converts the sparse label to one-hot label.- Parameters:
numClass- number of classes
-
-