Package | Description |
---|---|
org.tribuo.interop.onnx.extractors |
Provides feature extraction implementations which use ONNX models.
|
Modifier and Type | Method and Description |
---|---|
static BERTFeatureExtractor.OutputPooling |
BERTFeatureExtractor.OutputPooling.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static BERTFeatureExtractor.OutputPooling[] |
BERTFeatureExtractor.OutputPooling.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Constructor and Description |
---|
BERTFeatureExtractor(OutputFactory<T> outputFactory,
Path modelPath,
Path tokenizerPath,
BERTFeatureExtractor.OutputPooling pooling,
int maxLength,
boolean useCUDA)
Constructs a BERTFeatureExtractor.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.