dataeval.utils.onnx

Utilities for working with ONNX models.

These utilities help prepare ONNX models for embedding extraction by identifying and exposing intermediate layers that produce embeddings.

Functions

find_embedding_layer

Find the embedding layer name in an ONNX classification model.

to_encoding_model

Modify an ONNX model to output embeddings from an intermediate layer.