dataeval.utils.torch

PyTorch is the primary backend for metrics that require neural networks.

While these metrics can take in custom models, DataEval provides utility classes to create a seamless integration between custom models and DataEval’s metrics.

Submodules

models

Simple PyTorch model architectures used by DataEval.

trainer

Utility classes for training PyTorch models.