Coverage for packages / dqm-ml-pytorch / src / dqm_ml_pytorch / __init__.py: 100%

1"""DQM ML PyTorch package for deep learning-based data quality metrics. 

2 

3This package provides metric processors that use PyTorch models for 

4computing image embeddings and domain gap metrics. 

5 

6Classes: 

7 ImageEmbeddingProcessor: Extracts image embeddings using pre-trained CNNs. 

8 DomainGapProcessor: Computes statistical distances between datasets. 

9""" 

10 

11from dqm_ml_pytorch.domain_gap import DomainGapProcessor 

12from dqm_ml_pytorch.image_embedding import ImageEmbeddingProcessor 

13 

14__all__ = ["DomainGapProcessor", "ImageEmbeddingProcessor"]