Package org.tribuo.interop.onnx.extractors

Enum Class BERTFeatureExtractor.OutputPooling

java.lang.Object

java.lang.Enum<BERTFeatureExtractor.OutputPooling>

org.tribuo.interop.onnx.extractors.BERTFeatureExtractor.OutputPooling

All Implemented Interfaces:

Serializable, Comparable<BERTFeatureExtractor.OutputPooling>, Constable

Enclosing class:

BERTFeatureExtractor<T extends Output<T>>

public static enum BERTFeatureExtractor.OutputPooling
extends Enum<BERTFeatureExtractor.OutputPooling>

The type of output pooling to perform.

Nested Class Summary

Nested classes/interfaces inherited from class java.lang.Enum

Enum.EnumDesc<E extends Enum<E>>

Enum Constant Summary

Enum Constants

Enum Constant	Description
CLS	Returns the CLS embedding.
CLS_AND_MEAN	Takes the average of the token embeddings and the CLS token.
MEAN	Takes the average of all the token embeddings

Method Summary

Modifier and Type	Method	Description
static BERTFeatureExtractor.OutputPooling	valueOf(String name)	Returns the enum constant of this class with the specified name.
static BERTFeatureExtractor.OutputPooling[]	values()	Returns an array containing the constants of this enum class, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, describeConstable, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Details

CLS

public static final BERTFeatureExtractor.OutputPooling CLS

Returns the CLS embedding.

MEAN

public static final BERTFeatureExtractor.OutputPooling MEAN

Takes the average of all the token embeddings

CLS_AND_MEAN

public static final BERTFeatureExtractor.OutputPooling CLS_AND_MEAN

Takes the average of the token embeddings and the CLS token.

Method Details

values

public static BERTFeatureExtractor.OutputPooling[] values()

Returns an array containing the constants of this enum class, in the order they are declared.

Returns:

an array containing the constants of this enum class, in the order they are declared

valueOf

public static BERTFeatureExtractor.OutputPooling valueOf(String name)

Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

IllegalArgumentException - if this enum class has no constant with the specified name

NullPointerException - if the argument is null