dataeval 0.71.0__py3-none-any.whl → 0.71.1__py3-none-any.whl

dataeval/__init__.py

@@ -1,8 +1,9 @@
-__version__ = "0.71.0"
+__version__ = "0.71.1"
 
 from importlib.util import find_spec
 
 _IS_TORCH_AVAILABLE = find_spec("torch") is not None
+_IS_TORCHVISION_AVAILABLE = find_spec("torchvision") is not None
 _IS_TENSORFLOW_AVAILABLE = find_spec("tensorflow") is not None and find_spec("tensorflow_probability") is not None
 
 del find_spec

dataeval/_internal/detectors/ood/ae.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 
 from typing import Callable
 
-import keras
 import numpy as np
 import tensorflow as tf
+import tf_keras as keras
 from numpy.typing import ArrayLike
 
 from dataeval._internal.detectors.ood.base import OODBase, OODScoreOutput

dataeval/_internal/detectors/ood/aegmm.py

@@ -10,8 +10,8 @@ from __future__ import annotations
 
 from typing import Callable
 
-import keras
 import tensorflow as tf
+import tf_keras as keras
 from numpy.typing import ArrayLike
 
 from dataeval._internal.detectors.ood.base import OODGMMBase, OODScoreOutput

dataeval/_internal/detectors/ood/base.py

@@ -12,9 +12,9 @@ from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from typing import Callable, Literal, cast
 
-import keras
 import numpy as np
 import tensorflow as tf
+import tf_keras as keras
 from numpy.typing import ArrayLike, NDArray
 
 from dataeval._internal.interop import to_numpy

dataeval/_internal/detectors/ood/llr.py

@@ -11,12 +11,12 @@ from __future__ import annotations
 from functools import partial
 from typing import Callable
 
-import keras
 import numpy as np
 import tensorflow as tf
-from keras.layers import Input
-from keras.models import Model
+import tf_keras as keras
 from numpy.typing import ArrayLike, NDArray
+from tf_keras.layers import Input
+from tf_keras.models import Model
 
 from dataeval._internal.detectors.ood.base import OODBase, OODScoreOutput
 from dataeval._internal.interop import to_numpy

dataeval/_internal/detectors/ood/vae.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 
 from typing import Callable
 
-import keras
 import numpy as np
 import tensorflow as tf
+import tf_keras as keras
 from numpy.typing import ArrayLike
 
 from dataeval._internal.detectors.ood.base import OODBase, OODScoreOutput

dataeval/_internal/detectors/ood/vaegmm.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 
 from typing import Callable
 
-import keras
 import numpy as np
 import tensorflow as tf
+import tf_keras as keras
 from numpy.typing import ArrayLike
 
 from dataeval._internal.detectors.ood.base import OODGMMBase, OODScoreOutput

dataeval/_internal/detectors/outliers.py

@@ -147,7 +147,7 @@ class Outliers:
                 mask = _get_outlier_mask(values.astype(np.float64), self.outlier_method, self.outlier_threshold)
                 indices = np.flatnonzero(mask)
                 for i, value in zip(indices, values[mask]):
-                    flagged_images.setdefault(i, {}).update({stat: value})
+                    flagged_images.setdefault(int(i), {}).update({stat: value})
 
         return dict(sorted(flagged_images.items()))
 

dataeval/_internal/metrics/stats/datasetstats.py

@@ -162,10 +162,10 @@ def channelstats(
 
     >>> stats = channelstats(images)
     >>> print(stats.visualstats.darkness)
-    [0.02124 0.1213  0.2212  0.1013  0.1076  0.11383 0.2013  0.2076  0.2139
-     0.3013  0.3076  0.3137  0.4014  0.4075  0.4138  0.5015  0.508   0.5137
-     0.6016  0.6074  0.614   0.701   0.7075  0.714   0.8013  0.8076  0.814
-     0.9014  0.9077  0.914  ]
+    [0.07495 0.1748  0.275   0.1047  0.11096 0.1172  0.2047  0.2109  0.2172
+     0.3047  0.311   0.3171  0.4048  0.411   0.4172  0.505   0.5107  0.517
+     0.6045  0.611   0.617   0.7046  0.711   0.7173  0.8047  0.811   0.8174
+     0.905   0.911   0.917  ]
     """
     outputs = run_stats(images, bboxes, True, [PixelStatsProcessor, VisualStatsProcessor])
     return ChannelStatsOutput(*outputs)  # type: ignore

dataeval/_internal/metrics/stats/dimensionstats.py

@@ -57,16 +57,16 @@ class DimensionStatsProcessor(StatsProcessor[DimensionStatsOutput]):
     image_function_map = {
         "left": lambda x: x.box[0],
         "top": lambda x: x.box[1],
-        "width": lambda x: x.shape[-1],
-        "height": lambda x: x.shape[-2],
+        "width": lambda x: x.box[2] - x.box[0],
+        "height": lambda x: x.box[3] - x.box[1],
         "channels": lambda x: x.shape[-3],
-        "size": lambda x: np.prod(x.shape[-2:]),
-        "aspect_ratio": lambda x: x.shape[-1] / x.shape[-2],
+        "size": lambda x: (x.box[2] - x.box[0]) * (x.box[3] - x.box[1]),
+        "aspect_ratio": lambda x: (x.box[2] - x.box[0]) / (x.box[3] - x.box[1]),
         "depth": lambda x: get_bitdepth(x.image).depth,
         "center": lambda x: np.asarray([(x.box[0] + x.box[2]) / 2, (x.box[1] + x.box[3]) / 2]),
         "distance": lambda x: np.sqrt(
-            np.square(((x.box[0] + x.box[2]) / 2) - (x.width / 2))
-            + np.square(((x.box[1] + x.box[3]) / 2) - (x.height / 2))
+            np.square(((x.box[0] + x.box[2]) / 2) - (x.shape[-1] / 2))
+            + np.square(((x.box[1] + x.box[3]) / 2) - (x.shape[-2] / 2))
         ),
     }
 

dataeval/_internal/metrics/stats/visualstats.py

@@ -22,14 +22,14 @@ class VisualStatsOutput(BaseStatsOutput):
     ----------
     brightness : NDArray[np.float16]
         Brightness of the images
-    sharpness : NDArray[np.float16]
-        Blurriness of the images
     contrast : NDArray[np.float16]
         Image contrast ratio
     darkness : NDArray[np.float16]
         Darkness of the images
     missing : NDArray[np.float16]
         Percentage of the images with missing pixels
+    sharpness : NDArray[np.float16]
+        Sharpness of the images
     zeros : NDArray[np.float16]
         Percentage of the images with zero value pixels
     percentiles : NDArray[np.float16]
@@ -37,10 +37,10 @@ class VisualStatsOutput(BaseStatsOutput):
     """
 
     brightness: NDArray[np.float16]
-    sharpness: NDArray[np.float16]
     contrast: NDArray[np.float16]
     darkness: NDArray[np.float16]
     missing: NDArray[np.float16]
+    sharpness: NDArray[np.float16]
     zeros: NDArray[np.float16]
     percentiles: NDArray[np.float16]
 
@@ -49,25 +49,25 @@ class VisualStatsProcessor(StatsProcessor[VisualStatsOutput]):
     output_class = VisualStatsOutput
     cache_keys = ["percentiles"]
     image_function_map = {
-        "brightness": lambda x: x.get("percentiles")[-2],
-        "sharpness": lambda x: np.std(edge_filter(np.mean(x.image, axis=0))),
+        "brightness": lambda x: x.get("percentiles")[1],
         "contrast": lambda x: np.nan_to_num(
             (np.max(x.get("percentiles")) - np.min(x.get("percentiles"))) / np.mean(x.get("percentiles"))
         ),
-        "darkness": lambda x: x.get("percentiles")[1],
+        "darkness": lambda x: x.get("percentiles")[-2],
         "missing": lambda x: np.count_nonzero(np.isnan(np.sum(x.image, axis=0))) / np.prod(x.shape[-2:]),
+        "sharpness": lambda x: np.std(edge_filter(np.mean(x.image, axis=0))),
         "zeros": lambda x: np.count_nonzero(np.sum(x.image, axis=0) == 0) / np.prod(x.shape[-2:]),
         "percentiles": lambda x: np.nanpercentile(x.scaled, q=QUARTILES),
     }
     channel_function_map = {
-        "brightness": lambda x: x.get("percentiles")[:, -2],
-        "sharpness": lambda x: np.std(np.vectorize(edge_filter, signature="(m,n)->(m,n)")(x.image), axis=(1, 2)),
+        "brightness": lambda x: x.get("percentiles")[:, 1],
         "contrast": lambda x: np.nan_to_num(
             (np.max(x.get("percentiles"), axis=1) - np.min(x.get("percentiles"), axis=1))
             / np.mean(x.get("percentiles"), axis=1)
         ),
-        "darkness": lambda x: x.get("percentiles")[:, 1],
+        "darkness": lambda x: x.get("percentiles")[:, -2],
         "missing": lambda x: np.count_nonzero(np.isnan(x.image), axis=(1, 2)) / np.prod(x.shape[-2:]),
+        "sharpness": lambda x: np.std(np.vectorize(edge_filter, signature="(m,n)->(m,n)")(x.image), axis=(1, 2)),
         "zeros": lambda x: np.count_nonzero(x.image == 0, axis=(1, 2)) / np.prod(x.shape[-2:]),
         "percentiles": lambda x: np.nanpercentile(x.scaled, q=QUARTILES, axis=1).T,
     }
@@ -113,9 +113,10 @@ def visualstats(
 
     >>> results = visualstats(images)
     >>> print(results.brightness)
-    [0.0737 0.607  0.0713 0.1046 0.138  0.1713 0.2046 0.2379 0.2712 0.3047
-     0.338  0.3713 0.4045 0.438  0.4712 0.5044 0.538  0.5713 0.6045 0.638
-     0.6714 0.7046 0.738  0.7715 0.8047 0.838  0.871  0.905  0.938  0.971 ]
+    [0.02246 0.5557  0.06805 0.1014  0.1348  0.1681  0.2014  0.2347  0.268
+     0.3015  0.3347  0.3682  0.4014  0.4348  0.468   0.5015  0.5347  0.568
+     0.6016  0.635   0.668   0.701   0.735   0.768   0.8013  0.835   0.868
+     0.9014  0.9346  0.9683 ]
     >>> print(results.contrast)
     [2.041 1.332 1.293 1.279 1.272 1.268 1.265 1.263 1.261 1.26  1.259 1.258
      1.258 1.257 1.257 1.256 1.256 1.255 1.255 1.255 1.255 1.254 1.254 1.254

dataeval/_internal/models/tensorflow/autoencoder.py

@@ -12,9 +12,9 @@ from __future__ import annotations
 
 from typing import Callable, cast
 
-import keras
 import tensorflow as tf
-from keras.layers import (
+import tf_keras as keras
+from tf_keras.layers import (
     Dense,
     Flatten,
     Layer,

dataeval/_internal/models/tensorflow/losses.py

@@ -11,11 +11,11 @@ from __future__ import annotations
 from typing import Literal, cast
 
 import tensorflow as tf
-from keras.layers import Flatten
 from numpy.typing import NDArray
 from tensorflow_probability.python.distributions.mvn_diag import MultivariateNormalDiag
 from tensorflow_probability.python.distributions.mvn_tril import MultivariateNormalTriL
 from tensorflow_probability.python.stats import covariance
+from tf_keras.layers import Flatten
 
 from dataeval._internal.models.tensorflow.gmm import gmm_energy, gmm_params
 

dataeval/_internal/models/tensorflow/pixelcnn.py

@@ -13,9 +13,9 @@ from __future__ import annotations
 import functools
 import warnings
 
-import keras
 import numpy as np
 import tensorflow as tf
+import tf_keras as keras
 from tensorflow_probability.python.bijectors import bijector
 from tensorflow_probability.python.distributions import (
     categorical,

dataeval/_internal/models/tensorflow/trainer.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 
 from typing import Callable, Iterable, cast
 
-import keras
 import numpy as np
 import tensorflow as tf
+import tf_keras as keras
 from numpy.typing import NDArray
 
 

dataeval/_internal/models/tensorflow/utils.py

@@ -11,11 +11,13 @@ from __future__ import annotations
 import math
 from typing import Callable, Union, cast
 
-import keras as keras
 import numpy as np
 import tensorflow as tf
-from keras import Sequential
-from keras.layers import (
+import tf_keras as keras
+from numpy.typing import NDArray
+from tensorflow._api.v2.nn import relu, softmax, tanh
+from tf_keras import Sequential
+from tf_keras.layers import (
     Conv2D,
     Conv2DTranspose,
     Dense,
@@ -23,8 +25,6 @@ from keras.layers import (
     InputLayer,
     Reshape,
 )
-from numpy.typing import NDArray
-from tensorflow._api.v2.nn import relu, softmax, tanh
 
 from dataeval._internal.models.tensorflow.autoencoder import AE, AEGMM, VAE, VAEGMM
 from dataeval._internal.models.tensorflow.pixelcnn import PixelCNN

dataeval/utils/tensorflow/__init__.py

@@ -6,6 +6,12 @@ as well as constructors which allow for customization of the encoder, decoder an
 layers used by the model.
 """
 
+from dataeval import _IS_TENSORFLOW_AVAILABLE
+
 from . import loss, models, recon
 
-__all__ = ["loss", "models", "recon"]
+__all__ = []
+
+
+if _IS_TENSORFLOW_AVAILABLE:
+    __all__ = ["loss", "models", "recon"]

dataeval/utils/tensorflow/loss/__init__.py

@@ -1,3 +1,7 @@
+from dataeval import _IS_TENSORFLOW_AVAILABLE
 from dataeval._internal.models.tensorflow.losses import Elbo, LossGMM
 
-__all__ = ["Elbo", "LossGMM"]
+__all__ = []
+
+if _IS_TENSORFLOW_AVAILABLE:
+    __all__ += ["Elbo", "LossGMM"]

dataeval/utils/tensorflow/models/__init__.py

@@ -1,5 +1,9 @@
+from dataeval import _IS_TENSORFLOW_AVAILABLE
 from dataeval._internal.models.tensorflow.autoencoder import AE, AEGMM, VAE, VAEGMM
 from dataeval._internal.models.tensorflow.pixelcnn import PixelCNN
 from dataeval._internal.models.tensorflow.utils import create_model
 
-__all__ = ["create_model", "AE", "AEGMM", "PixelCNN", "VAE", "VAEGMM"]
+__all__ = []
+
+if _IS_TENSORFLOW_AVAILABLE:
+    __all__ += ["create_model", "AE", "AEGMM", "PixelCNN", "VAE", "VAEGMM"]

dataeval/utils/torch/__init__.py

@@ -5,8 +5,17 @@ While these metrics can take in custom models, DataEval provides utility classes
 to create a seamless integration between custom models and DataEval's metrics.
 """
 
+from dataeval import _IS_TORCH_AVAILABLE, _IS_TORCHVISION_AVAILABLE
 from dataeval._internal.utils import read_dataset
 
-from . import models, trainer
+__all__ = []
 
-__all__ = ["read_dataset", "models", "trainer"]
+if _IS_TORCH_AVAILABLE:
+    from . import models, trainer
+
+    __all__ += ["read_dataset", "models", "trainer"]
+
+if _IS_TORCHVISION_AVAILABLE:
+    from . import datasets
+
+    __all__ += ["datasets"]

dataeval/utils/torch/datasets/__init__.py

@@ -2,6 +2,11 @@
 Provide access to common Torch datasets used for computer vision
 """
 
-from dataeval._internal.datasets import CIFAR10, MNIST, VOCDetection
+from dataeval import _IS_TORCHVISION_AVAILABLE
 
-__all__ = ["CIFAR10", "MNIST", "VOCDetection"]
+__all__ = []
+
+if _IS_TORCHVISION_AVAILABLE:
+    from dataeval._internal.datasets import CIFAR10, MNIST, VOCDetection
+
+    __all__ += ["CIFAR10", "MNIST", "VOCDetection"]

dataeval/utils/torch/models/__init__.py

@@ -1,7 +1,11 @@
+from dataeval import _IS_TORCH_AVAILABLE
 from dataeval._internal.models.pytorch.autoencoder import (
     AriaAutoencoder,
     Decoder,
     Encoder,
 )
 
-__all__ = ["AriaAutoencoder", "Decoder", "Encoder"]
+__all__ = []
+
+if _IS_TORCH_AVAILABLE:
+    __all__ += ["AriaAutoencoder", "Decoder", "Encoder"]

dataeval/utils/torch/trainer/__init__.py

@@ -1,3 +1,7 @@
+from dataeval import _IS_TORCH_AVAILABLE
 from dataeval._internal.models.pytorch.autoencoder import AETrainer
 
-__all__ = ["AETrainer"]
+__all__ = []
+
+if _IS_TORCH_AVAILABLE:
+    __all__ += ["AETrainer"]

{dataeval-0.71.0.dist-info → dataeval-0.71.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dataeval
-Version: 0.71.0
+Version: 0.71.1
 Summary: DataEval provides a simple interface to characterize image data and its impact on model performance across classification and object-detection tasks
 Home-page: https://dataeval.ai/
 License: MIT
@@ -29,8 +29,9 @@ Requires-Dist: nvidia-cudnn-cu11 (>=8.6.0.163) ; extra == "tensorflow" or extra
 Requires-Dist: pillow (>=10.3.0)
 Requires-Dist: scikit-learn (>=1.5.0)
 Requires-Dist: scipy (>=1.10)
-Requires-Dist: tensorflow (>=2.14.1,<2.16) ; extra == "tensorflow" or extra == "all"
-Requires-Dist: tensorflow_probability (>=0.22.1,<0.24) ; extra == "tensorflow" or extra == "all"
+Requires-Dist: tensorflow (>=2.16) ; extra == "tensorflow" or extra == "all"
+Requires-Dist: tensorflow_probability (>=0.24) ; extra == "tensorflow" or extra == "all"
+Requires-Dist: tf-keras (>2.16) ; extra == "tensorflow" or extra == "all"
 Requires-Dist: torch (>=2.2.0) ; extra == "torch" or extra == "all"
 Requires-Dist: torchvision (>=0.17.0) ; extra == "torch" or extra == "all"
 Requires-Dist: tqdm

{dataeval-0.71.0.dist-info → dataeval-0.71.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-dataeval/__init__.py,sha256=3y4e1-DHeOSYpqdQcYoQC185-eYkOURrhDoBgUIquAg,555
+dataeval/__init__.py,sha256=Qm1rDTX_NyCAtZl2ilQ49v0j_zqnWhhVwIhe0cvrKjk,620
 dataeval/_internal/datasets.py,sha256=KbXSR-vOAzFamfXHRnI9mhhqUzEPyGpK47fZsirQn1I,14638
 dataeval/_internal/detectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dataeval/_internal/detectors/clusterer.py,sha256=srqTHzh9kIy7Ty4VYaptwuQlBh8emFeiEAeS_mYFKro,20750
@@ -12,13 +12,13 @@ dataeval/_internal/detectors/drift/uncertainty.py,sha256=Ot8L42AnFbkij4J3Tis7VzX
 dataeval/_internal/detectors/duplicates.py,sha256=wggaIl3uFxihNBQhPv5JcreZbhVaFKoMAJMv_9-aaHU,5324
 dataeval/_internal/detectors/merged_stats.py,sha256=okXGrqAgsqfANMxfIjiUQlZWlaIh5TVvIB9UPsOJZ7k,1351
 dataeval/_internal/detectors/ood/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dataeval/_internal/detectors/ood/ae.py,sha256=AIffh11UKZX-3oocDDp8-t-uDUm8aQKvrV0nIE3BLQo,2140
-dataeval/_internal/detectors/ood/aegmm.py,sha256=q2kRXZM5X0PoA10mRsi8Gh-W5FdFzEsnM1yDq3GFzn0,2518
-dataeval/_internal/detectors/ood/base.py,sha256=qYSmcN74x5-qL0_I7uNo4eQ8X8pr7M4cwjs2qvkJt5g,8561
-dataeval/_internal/detectors/ood/llr.py,sha256=VgZtMrMgI8zDVUzsqm2l4tqsULFIhdQeDd4lzdo_G7U,10217
-dataeval/_internal/detectors/ood/vae.py,sha256=iXEltu5pATWr42-28hZ3ARZavJrptLwUM5P9c8omA_s,3030
-dataeval/_internal/detectors/ood/vaegmm.py,sha256=ujp6UN0wpZcmPDPkVfTHZxgka5kuTOSzgXMmbKdmK2U,2947
-dataeval/_internal/detectors/outliers.py,sha256=VSHxfLUPj8VZTcPgQCqVLtpL88hZCCni_1JUfFamOrA,10201
+dataeval/_internal/detectors/ood/ae.py,sha256=OTcfvoiCdSFGaAAkejBKwwiHaHKB6sa01aW5fVBI1Bk,2152
+dataeval/_internal/detectors/ood/aegmm.py,sha256=7fRcTXfyUgYfcZOaa9GpGNNxAAp4sQ9zYowfs4s4420,2530
+dataeval/_internal/detectors/ood/base.py,sha256=jMMObJgPUZc2Vbql_UYNXvQAFO305TRhdVxk0YqfzJo,8573
+dataeval/_internal/detectors/ood/llr.py,sha256=wzWOeyqbD0WdXHRa1Qf-_3TbJYEfT6OuTQfcepPsbTM,10235
+dataeval/_internal/detectors/ood/vae.py,sha256=y_HP3tk7Clo3YG9hl-gke9_tJ4XW8x8sQlrC9ZtbVLw,3042
+dataeval/_internal/detectors/ood/vaegmm.py,sha256=SvdUKC8cVyEWfEGcczRmyA4SGJhbol0eDSDry1mZxII,2959
+dataeval/_internal/detectors/outliers.py,sha256=C7Iu66ze5KCCRQNc1TsqkFVKDFGfP4qjGMUv6RUpk-E,10206
 dataeval/_internal/interop.py,sha256=FLXJY-5hwJcKCtruyvaarqynXCMfcLbQSFvGnrWQDPo,1338
 dataeval/_internal/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dataeval/_internal/metrics/balance.py,sha256=8KwuR5HvytJtS1YW9KkNrCu2dPn_gP4FSbXrxQ-9kK8,6182
@@ -29,12 +29,12 @@ dataeval/_internal/metrics/diversity.py,sha256=ZSlq1KBvkMRVAvlznils2QmlPC73TTpHs
 dataeval/_internal/metrics/parity.py,sha256=eTjltNBJOTFH6T_t9V9-1EFr_U0vqlU642o3x2RWgz0,16527
 dataeval/_internal/metrics/stats/base.py,sha256=DRTaaFVtbH1M-wLO2NrtuAAXx699vlEjP9d2no72pM4,11066
 dataeval/_internal/metrics/stats/boxratiostats.py,sha256=Ac6nB41q43xHCJRDEXHNgsJF80VE8MpH8_kySxA84BE,6342
-dataeval/_internal/metrics/stats/datasetstats.py,sha256=vwJvb5nLvL17hKqO4ES-dEp6LELOT1w2P8dRWGyEjZI,6201
-dataeval/_internal/metrics/stats/dimensionstats.py,sha256=s9jM4MhIQPpLEhQw3mXOEijsmhmV7mLE0HEnWqqWLbQ,3848
+dataeval/_internal/metrics/stats/datasetstats.py,sha256=1H8Njtr27oNO8Hn3pwizAlOFkVe3QpbJb-RYk4dLKkY,6201
+dataeval/_internal/metrics/stats/dimensionstats.py,sha256=EIXrRia7OyB147WgAW2tqEwPMcCNWmSQidx5uQukSqQ,3915
 dataeval/_internal/metrics/stats/hashstats.py,sha256=I-aX-R0Rlvjwo7A5bjq3Bqs7-utTapnXB87z9TyC12w,2088
 dataeval/_internal/metrics/stats/labelstats.py,sha256=BNxI2flvKhSps2o4-TPbN9nf52ctatI2SuDZ07hah5E,4058
 dataeval/_internal/metrics/stats/pixelstats.py,sha256=_b0TdjHZwe2yj5Cdmz2IhbQP4LTnHI1qFlDgPV8fuCs,4420
-dataeval/_internal/metrics/stats/visualstats.py,sha256=2kwhvwBVDtRdF3hrM-Hd_SkRIbSB2fK80pFesJULXkI,4759
+dataeval/_internal/metrics/stats/visualstats.py,sha256=TdPwiehv0dY5HJmOOQk4_omfMd725NqOPG21A-q_t0I,4788
 dataeval/_internal/metrics/uap.py,sha256=RumSQey6vNoz9CtOG2_Inb-TurKJrAHqwhkyWBirxhk,2128
 dataeval/_internal/metrics/utils.py,sha256=vW3mQHjF0AvYlml27X5dZgd0YBk3zyBvvztLEfdRkvI,13475
 dataeval/_internal/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -43,12 +43,12 @@ dataeval/_internal/models/pytorch/autoencoder.py,sha256=nPyLjLZrPNla-lMnym3fUW-O
 dataeval/_internal/models/pytorch/blocks.py,sha256=pm2xwsDZjZJYXrhhiz8husvh2vHmrkFMSYEn-EDUD5Q,1354
 dataeval/_internal/models/pytorch/utils.py,sha256=Qgwym1PxGuwxbXCKUT-8r6Iyrxqm7x94oj45Vf5_CjE,1675
 dataeval/_internal/models/tensorflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dataeval/_internal/models/tensorflow/autoencoder.py,sha256=Ryn11jDbpZJOM5De-kMGPdbJBQEdwip6B20ajS8HqpE,10354
+dataeval/_internal/models/tensorflow/autoencoder.py,sha256=l-3utb7Rwq6_KiqOPBHnYxR7ngYhpIDFvbvNZc5UvNI,10369
 dataeval/_internal/models/tensorflow/gmm.py,sha256=QoEgbeax1GETqRmUF7A2ih9uFOZfFAjGzgH2ljExlAc,3669
-dataeval/_internal/models/tensorflow/losses.py,sha256=pZH5RnlM9R0RrBde9Lgq32muwAp7_PWc56Mu4u8RVvo,3976
-dataeval/_internal/models/tensorflow/pixelcnn.py,sha256=keI1gTNjBk18YD91Cp4exfuGYWU9lt-wapvhSazhcVs,48319
-dataeval/_internal/models/tensorflow/trainer.py,sha256=xNY0Iw7Qa1TnCuy9N1b77_VduFoW_BhbZjfQCxOVby4,4082
-dataeval/_internal/models/tensorflow/utils.py,sha256=l6jXKMWyQAEI4LpAONq95Xwr7CPgrs408ypf9TuNxkY,8732
+dataeval/_internal/models/tensorflow/losses.py,sha256=LavFmi9AWfE_HO4YxQ54kV8LZG5-UeCOhOlcPpxo-ic,3979
+dataeval/_internal/models/tensorflow/pixelcnn.py,sha256=ru4KF8CZHKbOpp-ZgDxuRdbcv_nTCs1i1H2lTMamL7Y,48331
+dataeval/_internal/models/tensorflow/trainer.py,sha256=LJ3t6Ud95cofKN-cgb5o5nDrYSFse7LSDOYIBkMgDJk,4094
+dataeval/_internal/models/tensorflow/utils.py,sha256=Uq6eUTEeUHGopL1_VBH656-Ue18v6WgiEUjmk8SMsc8,8741
 dataeval/_internal/output.py,sha256=qVbOi41dvfQICQ4uxysHPWBRKo1XR61kXHPL_vKOPm0,2545
 dataeval/_internal/utils.py,sha256=jo6bGJZAgyuZqRpAAC4gwhAHYE12316na19ZuFwMqes,1504
 dataeval/_internal/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -65,16 +65,16 @@ dataeval/metrics/estimators/__init__.py,sha256=4VFMKLPsJdaWiflf84bXGQ2k8ertFQ4WE
 dataeval/metrics/stats/__init__.py,sha256=AKlNelORMOM2OA9XIvwZ9nOn6dK6k-r-69ldEAuqgLA,1156
 dataeval/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dataeval/utils/__init__.py,sha256=cW_5-DIZG2OFRs3FVLOz0uCv4JWdaoVO7C9rOlR7ZEA,526
-dataeval/utils/tensorflow/__init__.py,sha256=XgjqrMtI00ERPPpdokbO1lDyc_H3CZ1TTqUXIj0B6PI,435
-dataeval/utils/tensorflow/loss/__init__.py,sha256=E9eB87LNh0o5nUCqssB027EXBsOfEayNHPcNW0QGFdA,101
-dataeval/utils/tensorflow/models/__init__.py,sha256=OVpmHF8itDcgOlfw6N9jr7IphZPbMJoiu7OdqYhU9fs,291
+dataeval/utils/tensorflow/__init__.py,sha256=sKRG3b_MLQUrAftkRZ17JyNZt6gjEguvTK83hO_IwRQ,530
+dataeval/utils/tensorflow/loss/__init__.py,sha256=s7tD_5dYWcNDmntGiEHhG7bVDsMAY1UO8FpQFe9cUns,195
+dataeval/utils/tensorflow/models/__init__.py,sha256=1R9Oi5DOYwT0W3JSEfoMsPOvhYFaKqKilwkrUifNnig,385
 dataeval/utils/tensorflow/recon/__init__.py,sha256=xe6gAQqK9tyAoDQTtaJAxIPK1humt5QzsG_9NPsqx58,116
-dataeval/utils/torch/__init__.py,sha256=bYUm-nNlNVU3bqDz7dQHFmaRWgLy3lLrD4cSDumDlxQ,373
-dataeval/utils/torch/datasets/__init__.py,sha256=S6C4OaxEjJJaIpHSZcZfkl4U5iS5YtZ9N5GYHqvbzvM,191
-dataeval/utils/torch/models/__init__.py,sha256=YnDnePYpRIKHyYn3F5qR1OObMSb-g0FGvI8X-uTB09E,162
-dataeval/utils/torch/trainer/__init__.py,sha256=Te-qElt8h-Zv8NN0r-VJOEdCPHTQ2yO3rd2MhRiZGZs,93
+dataeval/utils/torch/__init__.py,sha256=430fNKbqLByuGSeNhnoIJy3g9Z94ckZsAKWUZ15MVP4,575
+dataeval/utils/torch/datasets/__init__.py,sha256=94k7fMQdxYlQXDYouAHUgrQJ2oBwnvq4koFJpyhlUVA,292
+dataeval/utils/torch/models/__init__.py,sha256=q1BzoLHWA0uBXzT2glWJDrxVA1BN7xnkT2r_d-7Dlyw,246
+dataeval/utils/torch/trainer/__init__.py,sha256=hpcrlCCXPzb8b7FOzEAKqFy6Z7Zl4V_cx3yA7n3L1L4,177
 dataeval/workflows/__init__.py,sha256=VFeJyMhZxvj8WnU5Un32mwO8lNfBQOBjD9IdOqexnAE,320
-dataeval-0.71.0.dist-info/LICENSE.txt,sha256=Kpzcfobf1HlqafF-EX6dQLw9TlJiaJzfgvLQFukyXYw,1060
-dataeval-0.71.0.dist-info/METADATA,sha256=b1faVcCXttUUf9VQ1-TBXTRg6Kv_OEj1nbMNbpo4B5g,4522
-dataeval-0.71.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-dataeval-0.71.0.dist-info/RECORD,,
+dataeval-0.71.1.dist-info/LICENSE.txt,sha256=Kpzcfobf1HlqafF-EX6dQLw9TlJiaJzfgvLQFukyXYw,1060
+dataeval-0.71.1.dist-info/METADATA,sha256=j1HnzvkOSLR-D6debnFFLlp2vgue8ueFQvjYlrX75cw,4580
+dataeval-0.71.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+dataeval-0.71.1.dist-info/RECORD,,