org.tribuo.interop.onnx.extractors

Enum BERTFeatureExtractor.OutputPooling

java.lang.Object

java.lang.Enum<BERTFeatureExtractor.OutputPooling>

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

All Implemented Interfaces:

Serializable, Comparable<BERTFeatureExtractor.OutputPooling>

Enclosing class:

BERTFeatureExtractor<T extends Output<T>>

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

The type of output pooling to perform.

Enum Constant Summary

Enum Constants

Enum Constant and 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 and Description
static BERTFeatureExtractor.OutputPooling	valueOf(String name) Returns the enum constant of this type with the specified name.
static BERTFeatureExtractor.OutputPooling[]	values() Returns an array containing the constants of this enum type, in the order they are declared.

Methods inherited from class java.lang.Enum

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

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Detail

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 Detail

values

public static BERTFeatureExtractor.OutputPooling[] values()

Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows:

for (BERTFeatureExtractor.OutputPooling c : BERTFeatureExtractor.OutputPooling.values())
    System.out.println(c);

Returns:

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

valueOf

public static BERTFeatureExtractor.OutputPooling valueOf(String name)

Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (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 type has no constant with the specified name

NullPointerException - if the argument is null