Coverage for /opt/hostedtoolcache/Python/3.10.20/x64/lib/python3.10/site-packages/equine/equine_output.py: 100%

1# Copyright 2024, MASSACHUSETTS INSTITUTE OF TECHNOLOGY 

2# Subject to FAR 52.227-11 – Patent Rights – Ownership by the Contractor (May 2014). 

3# SPDX-License-Identifier: MIT 

4from dataclasses import dataclass 

5 

6from torch import Tensor 

7 

8 

9@dataclass 

10class EquineOutput: 

11 """ 

12 Output object containing prediction probabilities, OOD scores, and embeddings, which can be used for visualization. 

13 

14 Attributes 

15 ---------- 

16 classes : Tensor 

17 Tensor of predicted class probabilities. 

18 ood_scores : Tensor 

19 Tensor of out-of-distribution (OOD) scores. 

20 embeddings : Tensor 

21 Tensor of embeddings produced by the model. 

22 """ 

23 

24 classes: Tensor 

25 ood_scores: Tensor 

26 embeddings: Tensor