Package ai.djl.modality.cv.transform

Class OneHot

java.lang.Object

ai.djl.modality.cv.transform.OneHot

All Implemented Interfaces:

Transform

public class OneHot
extends Object
implements Transform

A Transform that converts the labels NDArray to one-hot labels.

Constructor Summary

Constructors

Constructor	Description
OneHot(int numClass)	Creates a OneHot Transform that converts the sparse label to one-hot label.

Method Summary

Modifier and Type	Method	Description
NDArray	transform(NDArray array)	Applies the Transform to the given NDArray.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

OneHot

public OneHot(int numClass)

Creates a OneHot Transform that converts the sparse label to one-hot label.

Parameters:

numClass - number of classes

Method Details

transform

public NDArray transform(NDArray array)

Applies the Transform to the given NDArray.

Specified by:

transform in interface Transform

Parameters:

array - the NDArray on which the Transform is applied

Returns:

the output of the Transform