a  dS@sddlmZmZmZmZmZddlmZddlm Z ddl m Z m Z ddl mZddlmZddlmZddlmZmZesd gZGd d d eZd S) )AnyListOptionalSequenceUnion)Tensor)Literal)_cosine_similarity_compute_cosine_similarity_update)Metric) dim_zero_cat)_MATPLOTLIB_AVAILABLE)_AX_TYPE_PLOT_OUT_TYPECosineSimilarity.plotcseZdZUdZdZeed<dZeed<dZeed<dZ e ed<d Z e ed <e e ed <e e ed <dededdfdd Ze e ddddZe dddZdeee ee feeedddZZS)CosineSimilarityaCompute the `Cosine Similarity`_. .. math:: cos_{sim}(x,y) = \frac{x \cdot y}{||x|| \cdot ||y||} = \frac{\sum_{i=1}^n x_i y_i}{\sqrt{\sum_{i=1}^n x_i^2}\sqrt{\sum_{i=1}^n y_i^2}} where :math:`y` is a tensor of target values, and :math:`x` is a tensor of predictions. As input to ``forward`` and ``update`` the metric accepts the following input: - ``preds`` (:class:`~torch.Tensor`): Predicted float tensor with shape ``(N,d)`` - ``target`` (:class:`~torch.Tensor`): Ground truth float tensor with shape ``(N,d)`` As output of ``forward`` and ``compute`` the metric returns the following output: - ``cosine_similarity`` (:class:`~torch.Tensor`): A float tensor with the cosine similarity Args: reduction: how to reduce over the batch dimension using 'sum', 'mean' or 'none' (taking the individual scores) kwargs: Additional keyword arguments, see :ref:`Metric kwargs` for more info. Example: >>> from torch import tensor >>> from torchmetrics.regression import CosineSimilarity >>> target = tensor([[0, 1], [1, 1]]) >>> preds = tensor([[0, 1], [0, 1]]) >>> cosine_similarity = CosineSimilarity(reduction = 'mean') >>> cosine_similarity(preds, target) tensor(0.8536) Tis_differentiablehigher_is_betterFfull_state_updategplot_lower_boundg?plot_upper_boundpredstargetsum)meanrnoneNN) reductionkwargsreturnc s\tjfi|d}||vr2td|d|||_|jdgdd|jdgdddS)N)rrrNz+Expected argument `reduction` to be one of z but got rcat)Zdist_reduce_fxr)super__init__ ValueErrorrZ add_state)selfrrZallowed_reduction __class__r/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/regression/cosine_similarity.pyr!FszCosineSimilarity.__init__)rrrcCs*t||\}}|j||j|dS)z2Update metric states with predictions and targets.N)r rappendrr#rrr&r&r'updateTs zCosineSimilarity.update)rcCs"t|j}t|j}t|||jS)zCompute metric.)r rrr rr)r&r&r'compute[s  zCosineSimilarity.compute)valaxrcCs |||S)aPlot a single or multiple values from the metric. Args: val: Either a single result from calling `metric.forward` or `metric.compute` or a list of these results. If no value is provided, will automatically call `metric.compute` and plot that result. ax: An matplotlib axis object. If provided will add plot to that axis Returns: Figure and Axes object Raises: ModuleNotFoundError: If `matplotlib` is not installed .. plot:: :scale: 75 >>> from torch import randn >>> # Example plotting a single value >>> from torchmetrics.regression import CosineSimilarity >>> metric = CosineSimilarity() >>> metric.update(randn(10,), randn(10,)) >>> fig_, ax_ = metric.plot() .. plot:: :scale: 75 >>> from torch import randn >>> # Example plotting multiple values >>> from torchmetrics.regression import CosineSimilarity >>> metric = CosineSimilarity() >>> values = [] >>> for _ in range(10): ... values.append(metric(randn(10,), randn(10,))) >>> fig, ax = metric.plot(values) )Z_plot)r#r,r-r&r&r'plotas(r)r)NN)__name__ __module__ __qualname____doc__rbool__annotations__rrrfloatrrrrrr!r*r+rrrrrr. __classcell__r&r&r$r'rs*        rN)typingrrrrrZtorchrZtyping_extensionsrZ4torchmetrics.functional.regression.cosine_similarityr r Ztorchmetrics.metricr Ztorchmetrics.utilities.datar Ztorchmetrics.utilities.importsr Ztorchmetrics.utilities.plotrrZ__doctest_skip__rr&r&r&r's