Package org.tribuo.interop.onnx.extractors

Provides feature extraction implementations which use ONNX models.

See: Description

Class Summary

Class	Description
BERTFeatureExtractor<T extends Output<T>>	Builds examples and sequence examples using features from BERT.
BERTFeatureExtractor.BERTFeatureExtractorOptions	CLI options for running BERT.

Enum Summary

Enum	Description
BERTFeatureExtractor.OutputPooling	The type of output pooling to perform.

Package org.tribuo.interop.onnx.extractors Description

Provides feature extraction implementations which use ONNX models.

These are mainly for text or images, e.g., BERTFeatureExtractor which uses an ONNX format BERT model to provide feature vectors based on the BERT embeddings.