compositional-explanations 0.0.3.1__tar.gz → 0.0.3.2__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (21) hide show
  1. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/LICENSE +0 -0
  2. {compositional_explanations-0.0.3.1/src/compositional_explanations.egg-info → compositional_explanations-0.0.3.2}/PKG-INFO +1 -1
  3. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/README.md +0 -0
  4. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/pyproject.toml +1 -1
  5. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/metrics.py +8 -6
  6. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2/src/compositional_explanations.egg-info}/PKG-INFO +1 -1
  7. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/package_utils/constants.py +0 -0
  8. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/package_utils/general_utils.py +0 -0
  9. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/package_utils/mask_utils.py +0 -0
  10. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/package_utils/optimal_utils.py +0 -0
  11. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/setup.cfg +0 -0
  12. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/beam.py +0 -0
  13. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/formula.py +0 -0
  14. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/optimal.py +0 -0
  15. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/optimal_sample_heuristic.py +0 -0
  16. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/optimal_sum_heuristic.py +0 -0
  17. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations/path_heuristic.py +0 -0
  18. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations.egg-info/SOURCES.txt +0 -0
  19. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations.egg-info/dependency_links.txt +0 -0
  20. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations.egg-info/requires.txt +0 -0
  21. {compositional_explanations-0.0.3.1 → compositional_explanations-0.0.3.2}/src/compositional_explanations.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: compositional_explanations
3
- Version: 0.0.3.1
3
+ Version: 0.0.3.2
4
4
  Summary: Package to compute Compositional Explanations both in their optimal and beam format.
5
5
  Author-email: Biagio La Rosa <bilarosa@ucsc.edu>
6
6
  License: MIT
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "compositional_explanations"
7
- version = "0.0.3.1"
7
+ version = "0.0.3.2"
8
8
  authors = [
9
9
  { name="Biagio La Rosa", email="bilarosa@ucsc.edu" },
10
10
  ]
@@ -43,17 +43,19 @@ def iou(vector1, vector2):
43
43
  return score
44
44
 
45
45
 
46
+
46
47
  def compute_iou_from_masks(*, formula, masks, bitmaps):
47
- """Compute the intersection over union between two vectors.
48
+ """Compute the IoU between a formula mask and a bitmap.
48
49
  Args:
49
- vector1 (torch.Tensor): A tensor of shape (N, H, W) where N is the
50
- number of sample.
51
- vector2 (torch.Tensor): A tensor of shape (N, H, W) where N is the
52
- number of sample.
50
+ formula (str): A string representing the formula.
51
+ masks (dict): A dictionary mapping concepts to their corresponding binary masks.
52
+ bitmaps (torch.Tensor): A tensor of shape (N, H, W) where N is the number of samples.
53
53
  Returns:
54
- iou (float): The intersection over union between the two vectors.
54
+ iou (float): The intersection over union between the formula mask and the bitmap.
55
55
  """
56
56
  formula_mask = mask_utils.get_formula_mask(formula, masks)
57
+ if formula_mask.device != bitmaps.device:
58
+ formula_mask = formula_mask.to(bitmaps.device)
57
59
  intersection = torch.count_nonzero(formula_mask & bitmaps).item()
58
60
  v1_size = compute_hits(formula_mask).item()
59
61
  v2_size = compute_hits(bitmaps).item()
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: compositional_explanations
3
- Version: 0.0.3.1
3
+ Version: 0.0.3.2
4
4
  Summary: Package to compute Compositional Explanations both in their optimal and beam format.
5
5
  Author-email: Biagio La Rosa <bilarosa@ucsc.edu>
6
6
  License: MIT