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(model_path)

Find the embedding layer name in an ONNX classification model.

to_encoding_model(…)

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