Coverage for  / opt / hostedtoolcache / Python / 3.10.19 / 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 

5from torch import Tensor 

6 

7 

8@dataclass 

9class EquineOutput: 

10 """ 

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

12 

13 Attributes 

14 ---------- 

15 classes : Tensor 

16 Tensor of predicted class probabilities. 

17 ood_scores : Tensor 

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

19 embeddings : Tensor 

20 Tensor of embeddings produced by the model. 

21 """ 

22 

23 classes: Tensor 

24 ood_scores: Tensor 

25 embeddings: Tensor