tlc.client.torch.metrics.metrics_collectors#

Implementations of metrics collectors in PyTorch.

Modules and Packages#

Module

Description

classification_metrics_collector

Collect loss and predictions for classification problems.

segmentation_metrics_collector

Collects metrics for segmentation problems.

embeddings_metrics_collector

Extract embeddings from PyTorch models.

bounding_box_metrics_collector

Collect metrics for bounding box predictions.

metrics_collector_base

Base class for all metrics collectors in PyTorch.

functional_metrics_collector

Use a custom function to collect metrics.