Package org.tribuo.interop.onnx.extractors

package org.tribuo.interop.onnx.extractors

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.

Related Packages

Package	Description
org.tribuo.interop.onnx	This package contains a Tribuo wrapper around ONNX Runtime.

Class	Description
BERTFeatureExtractor<T extends Output<T>>	Builds examples and sequence examples using features from BERT.
BERTFeatureExtractor.BERTFeatureExtractorOptions	CLI options for running BERT.
BERTFeatureExtractor.OutputPooling	The type of output pooling to perform.