





from sklearn.metrics import precision_score

import datasets


_UpperCamelCase       :				Any           =			"""\nPrecision is the fraction of correctly labeled positive examples out of all of the examples that were labeled as positive. It is computed via the equation:\nPrecision = TP / (TP + FP)\nwhere TP is the True positives (i.e. the examples correctly labeled as positive) and FP is the False positive examples (i.e. the examples incorrectly labeled as positive).\n"""


_UpperCamelCase       :				str           =			"""\nArgs:\n    predictions (`list` of `int`): Predicted class labels.\n    references (`list` of `int`): Actual class labels.\n    labels (`list` of `int`): The set of labels to include when `average` is not set to `\'binary\'`. If `average` is `None`, it should be the label order. Labels present in the data can be excluded, for example to calculate a multiclass average ignoring a majority negative class. Labels not present in the data will result in 0 components in a macro average. For multilabel targets, labels are column indices. By default, all labels in `predictions` and `references` are used in sorted order. Defaults to None.\n    pos_label (`int`): The class to be considered the positive class, in the case where `average` is set to `binary`. Defaults to 1.\n    average (`string`): This parameter is required for multiclass/multilabel targets. If set to `None`, the scores for each class are returned. Otherwise, this determines the type of averaging performed on the data. Defaults to `\'binary\'`.\n\n        - \'binary\': Only report results for the class specified by `pos_label`. This is applicable only if the classes found in `predictions` and `references` are binary.\n        - \'micro\': Calculate metrics globally by counting the total true positives, false negatives and false positives.\n        - \'macro\': Calculate metrics for each label, and find their unweighted mean. This does not take label imbalance into account.\n        - \'weighted\': Calculate metrics for each label, and find their average weighted by support (the number of true instances for each label). This alters `\'macro\'` to account for label imbalance. This option can result in an F-score that is not between precision and recall.\n        - \'samples\': Calculate metrics for each instance, and find their average (only meaningful for multilabel classification).\n    sample_weight (`list` of `float`): Sample weights Defaults to None.\n    zero_division (`int` or `string`): Sets the value to return when there is a zero division. Defaults to \'warn\'.\n\n        - 0: Returns 0 when there is a zero division.\n        - 1: Returns 1 when there is a zero division.\n        - \'warn\': Raises warnings and then returns 0 when there is a zero division.\n\nReturns:\n    precision (`float` or `array` of `float`): Precision score or list of precision scores, depending on the value passed to `average`. Minimum possible value is 0. Maximum possible value is 1. Higher values indicate that fewer negative examples were incorrectly labeled as positive, which means that, generally, higher scores are better.\n\nExamples:\n\n    Example 1-A simple binary example\n        >>> precision_metric = datasets.load_metric(\"precision\")\n        >>> results = precision_metric.compute(references=[0, 1, 0, 1, 0], predictions=[0, 0, 1, 1, 0])\n        >>> print(results)\n        {\'precision\': 0.5}\n\n    Example 2-The same simple binary example as in Example 1, but with `pos_label` set to `0`.\n        >>> precision_metric = datasets.load_metric(\"precision\")\n        >>> results = precision_metric.compute(references=[0, 1, 0, 1, 0], predictions=[0, 0, 1, 1, 0], pos_label=0)\n        >>> print(round(results[\'precision\'], 2))\n        0.67\n\n    Example 3-The same simple binary example as in Example 1, but with `sample_weight` included.\n        >>> precision_metric = datasets.load_metric(\"precision\")\n        >>> results = precision_metric.compute(references=[0, 1, 0, 1, 0], predictions=[0, 0, 1, 1, 0], sample_weight=[0.9, 0.5, 3.9, 1.2, 0.3])\n        >>> print(results)\n        {\'precision\': 0.23529411764705882}\n\n    Example 4-A multiclass example, with different values for the `average` input.\n        >>> predictions = [0, 2, 1, 0, 0, 1]\n        >>> references = [0, 1, 2, 0, 1, 2]\n        >>> results = precision_metric.compute(predictions=predictions, references=references, average=\'macro\')\n        >>> print(results)\n        {\'precision\': 0.2222222222222222}\n        >>> results = precision_metric.compute(predictions=predictions, references=references, average=\'micro\')\n        >>> print(results)\n        {\'precision\': 0.3333333333333333}\n        >>> results = precision_metric.compute(predictions=predictions, references=references, average=\'weighted\')\n        >>> print(results)\n        {\'precision\': 0.2222222222222222}\n        >>> results = precision_metric.compute(predictions=predictions, references=references, average=None)\n        >>> print([round(res, 2) for res in results[\'precision\']])\n        [0.67, 0.0, 0.0]\n"""


_UpperCamelCase       :				str           =			"""\n@article{scikit-learn,\n    title={Scikit-learn: Machine Learning in {P}ython},\n    author={Pedregosa, F. and Varoquaux, G. and Gramfort, A. and Michel, V.\n    and Thirion, B. and Grisel, O. and Blondel, M. and Prettenhofer, P.\n    and Weiss, R. and Dubourg, V. and Vanderplas, J. and Passos, A. and\n    Cournapeau, D. and Brucher, M. and Perrot, M. and Duchesnay, E.},\n    journal={Journal of Machine Learning Research},\n    volume={12},\n    pages={2825--2830},\n    year={2011}\n}\n"""







@datasets.utils.file_utils.add_start_docstrings(_DESCRIPTION						,       _KWARGS_DESCRIPTION)
class 		_lowerCAmelCase(		datasets.Metric):

  """simple docstring"""







  def 						SCREAMING_SNAKE_CASE__			(   self      )->   Optional[int]:
     return datasets.MetricInfo(
         description=_DESCRIPTION    ,							citation=_CITATION    ,							inputs_description=_KWARGS_DESCRIPTION    ,							features=datasets.Features(
             {
                 '''predictions''': datasets.Sequence(datasets.Value('''int32'''      )      ),
                 '''references''': datasets.Sequence(datasets.Value('''int32'''      )      ),
             }
             if self.config_name == '''multilabel'''
             else {
                 '''predictions''': datasets.Value('''int32'''      ),
                 '''references''': datasets.Value('''int32'''      ),
             }      )    ,							reference_urls=['''https://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_score.html''']    ,							)





  def 						SCREAMING_SNAKE_CASE__			(   self    ,							UpperCAmelCase    ,							UpperCAmelCase    ,							UpperCAmelCase=None    ,							UpperCAmelCase=1    ,							UpperCAmelCase="binary"    ,							UpperCAmelCase=None    ,							UpperCAmelCase="warn"    ,							)->   Any:
     __A    =  precision_score(
         lowerCamelCase_    ,							lowerCamelCase_    ,							labels=lowerCamelCase_    ,							pos_label=lowerCamelCase_    ,							average=lowerCamelCase_    ,							sample_weight=lowerCamelCase_    ,							zero_division=lowerCamelCase_    ,							)
     return {"precision": float(lowerCamelCase_      ) if score.size == 1 else score}


"""simple docstring"""







from dataclasses import dataclass
from typing import Any, Dict, Optional

import torch
import torch.nn.functional as F
from torch import nn

from ..configuration_utils import ConfigMixin, register_to_config
from ..models.embeddings import ImagePositionalEmbeddings
from ..utils import BaseOutput, deprecate
from .attention import BasicTransformerBlock
from .embeddings import PatchEmbed
from .modeling_utils import ModelMixin



@dataclass
class 			_lowerCamelCase      (__lowerCamelCase  ):
				_snake_case  =	42



class 			_lowerCamelCase      (__lowerCamelCase    ,				__lowerCamelCase  ):







				@register_to_config
				def __init__(			self     :							Dict      ,       lowerCamelCase_     :							int = 1_6      ,       lowerCamelCase_     :							int = 8_8      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							int = 1      ,       lowerCamelCase_     :							float = 0.0      ,       lowerCamelCase_     :							int = 3_2      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							bool = False      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							str = "geglu"      ,       lowerCamelCase_     :							Optional[int] = None      ,       lowerCamelCase_     :							bool = False      ,       lowerCamelCase_     :							bool = False      ,       lowerCamelCase_     :							bool = False      ,       lowerCamelCase_     :							str = "layer_norm"      ,       lowerCamelCase_     :							bool = True      ,       ):

					"""simple docstring"""






					super().__init__()
					_lowercase			:     Tuple      					=				use_linear_projection
					_lowercase			:     List[Any]      					=				num_attention_heads
					_lowercase			:     Optional[int]      					=				attention_head_dim
					_lowercase			:     Union[str, Any]      					=				num_attention_heads * attention_head_dim

					# 1. Transformer2DModel can process both standard continuous images of shape `(batch_size, num_channels, width, height)` as well as quantized image embeddings of shape `(batch_size, num_image_vectors)`
					# Define whether input is continuous or discrete depending on configuration
					_lowercase			:     Dict      					=				(in_channels is not None) and (patch_size is None)
					_lowercase			:     Any      					=				num_vector_embeds is not None
					_lowercase			:     List[Any]      					=				in_channels is not None and patch_size is not None

					if norm_type == "layer_norm" and num_embeds_ada_norm is not None:
						_lowercase			:     Tuple      					=				(
						    F'''The configuration file of this model: {self.__class__} is outdated. `norm_type` is either not set or'''
						    ' incorrectly set to `\'layer_norm\'`.Make sure to set `norm_type` to `\'ada_norm\'` in the config.'
						    ' Please make sure to update the config accordingly as leaving `norm_type` might led to incorrect'
						    ' results in future versions. If you have downloaded this checkpoint from the Hugging Face Hub, it'
						    ' would be very nice if you could open a Pull request for the `transformer/config.json` file'
						)
						deprecate('norm_type!=num_embeds_ada_norm'      ,       '1.0.0'      ,       lowerCamelCase_      ,       standard_warn=lowerCamelCase_    )
						_lowercase			:     Dict      					=				'ada_norm'

					if self.is_input_continuous and self.is_input_vectorized:
						raise ValueError(
						    F'''Cannot define both `in_channels`: {in_channels} and `num_vector_embeds`: {num_vector_embeds}. Make'''
						    ' sure that either `in_channels` or `num_vector_embeds` is None.'    )
					elif self.is_input_vectorized and self.is_input_patches:
						raise ValueError(
						    F'''Cannot define both `num_vector_embeds`: {num_vector_embeds} and `patch_size`: {patch_size}. Make'''
						    ' sure that either `num_vector_embeds` or `num_patches` is None.'    )
					elif not self.is_input_continuous and not self.is_input_vectorized and not self.is_input_patches:
						raise ValueError(
						    F'''Has to define `in_channels`: {in_channels}, `num_vector_embeds`: {num_vector_embeds}, or patch_size:'''
						    F''' {patch_size}. Make sure that `in_channels`, `num_vector_embeds` or `num_patches` is not None.'''    )

					# 2. Define input layers
					if self.is_input_continuous:
						_lowercase			:     int      					=				in_channels

						_lowercase			:     List[Any]      					=				torch.nn.GroupNorm(num_groups=lowerCamelCase_      ,       num_channels=lowerCamelCase_      ,       eps=1E-6      ,       affine=lowerCamelCase_    )
						if use_linear_projection:
							_lowercase			:     Any      					=				nn.Linear(lowerCamelCase_      ,       lowerCamelCase_    )
						else:
							_lowercase			:     List[str]      					=				nn.Convad(lowerCamelCase_      ,       lowerCamelCase_      ,       kernel_size=1      ,       stride=1      ,       padding=0    )
					elif self.is_input_vectorized:
						assert sample_size is not None, "Transformer2DModel over discrete input must provide sample_size"
						assert num_vector_embeds is not None, "Transformer2DModel over discrete input must provide num_embed"

						_lowercase			:     int      					=				sample_size
						_lowercase			:     str      					=				sample_size
						_lowercase			:     int      					=				num_vector_embeds
						_lowercase			:     List[str]      					=				self.height * self.width

						_lowercase			:     Optional[int]      					=				ImagePositionalEmbeddings(
						    num_embed=lowerCamelCase_      ,       embed_dim=lowerCamelCase_      ,       height=self.height      ,       width=self.width    )
					elif self.is_input_patches:
						assert sample_size is not None, "Transformer2DModel over patched input must provide sample_size"

						_lowercase			:     Optional[Any]      					=				sample_size
						_lowercase			:     Optional[int]      					=				sample_size

						_lowercase			:     str      					=				patch_size
						_lowercase			:     Union[str, Any]      					=				PatchEmbed(
						    height=lowerCamelCase_      ,       width=lowerCamelCase_      ,       patch_size=lowerCamelCase_      ,       in_channels=lowerCamelCase_      ,       embed_dim=lowerCamelCase_      ,       )

					# 3. Define transformers blocks
					_lowercase			:     Any      					=				nn.ModuleList(
					    [
					        BasicTransformerBlock(
					            lowerCamelCase_      ,       lowerCamelCase_      ,       lowerCamelCase_      ,       dropout=lowerCamelCase_      ,       cross_attention_dim=lowerCamelCase_      ,       activation_fn=lowerCamelCase_      ,       num_embeds_ada_norm=lowerCamelCase_      ,       attention_bias=lowerCamelCase_      ,       only_cross_attention=lowerCamelCase_      ,       upcast_attention=lowerCamelCase_      ,       norm_type=lowerCamelCase_      ,       norm_elementwise_affine=lowerCamelCase_      ,       )
					        for d in range(lowerCamelCase_    )
					    ]    )

					# 4. Define output layers
					_lowercase			:     Dict      					=				in_channels if out_channels is None else out_channels
					if self.is_input_continuous:
						# TODO: should use out_channels for continuous projections
						if use_linear_projection:
							_lowercase			:     str      					=				nn.Linear(lowerCamelCase_      ,       lowerCamelCase_    )
						else:
							_lowercase			:     List[Any]      					=				nn.Convad(lowerCamelCase_      ,       lowerCamelCase_      ,       kernel_size=1      ,       stride=1      ,       padding=0    )
					elif self.is_input_vectorized:
						_lowercase			:     Dict      					=				nn.LayerNorm(lowerCamelCase_    )
						_lowercase			:     Union[str, Any]      					=				nn.Linear(lowerCamelCase_      ,       self.num_vector_embeds - 1    )
					elif self.is_input_patches:
						_lowercase			:     Dict      					=				nn.LayerNorm(lowerCamelCase_      ,       elementwise_affine=lowerCamelCase_      ,       eps=1E-6    )
						_lowercase			:     Union[str, Any]      					=				nn.Linear(lowerCamelCase_      ,       2 * inner_dim    )
						_lowercase			:     List[Any]      					=				nn.Linear(lowerCamelCase_      ,       patch_size * patch_size * self.out_channels    )







				def 				__UpperCAmelCase       (			self     :							Optional[Any]      ,       lowerCamelCase_     :							torch.Tensor      ,       lowerCamelCase_     :							Optional[torch.Tensor] = None      ,       lowerCamelCase_     :							Optional[torch.LongTensor] = None      ,       lowerCamelCase_     :							Optional[torch.LongTensor] = None      ,       lowerCamelCase_     :							Dict[str, Any] = None      ,       lowerCamelCase_     :							Optional[torch.Tensor] = None      ,       lowerCamelCase_     :							Optional[torch.Tensor] = None      ,       lowerCamelCase_     :							bool = True      ,       ):

					"""simple docstring"""






					if attention_mask is not None and attention_mask.ndim == 2:
						# assume that mask is expressed as:
						#   (1 = keep,      0 = discard)
						# convert mask into a bias that can be added to attention scores:
						#       (keep = +0,     discard = -10000.0)
						_lowercase			:     Optional[Any]      					=				(1 - attention_mask.to(hidden_states.dtype    )) * -1_0000.0
						_lowercase			:     Optional[Any]      					=				attention_mask.unsqueeze(1    )

					# convert encoder_attention_mask to a bias the same way we do for attention_mask
					if encoder_attention_mask is not None and encoder_attention_mask.ndim == 2:
						_lowercase			:     List[Any]      					=				(1 - encoder_attention_mask.to(hidden_states.dtype    )) * -1_0000.0
						_lowercase			:     Any      					=				encoder_attention_mask.unsqueeze(1    )

					# 1. Input
					if self.is_input_continuous:
						_lowercase					,   _lowercase					,   _lowercase					,   _lowercase			:     List[Any]      					=				hidden_states.shape
						_lowercase			:     int      					=				hidden_states

						_lowercase			:     Any      					=				self.norm(lowerCamelCase_    )
						if not self.use_linear_projection:
							_lowercase			:     Any      					=				self.proj_in(lowerCamelCase_    )
							_lowercase			:     Any      					=				hidden_states.shape[1]
							_lowercase			:     List[Any]      					=				hidden_states.permute(0      ,       2      ,       3      ,       1    ).reshape(lowerCamelCase_      ,       height * width      ,       lowerCamelCase_    )
						else:
							_lowercase			:     Any      					=				hidden_states.shape[1]
							_lowercase			:     int      					=				hidden_states.permute(0      ,       2      ,       3      ,       1    ).reshape(lowerCamelCase_      ,       height * width      ,       lowerCamelCase_    )
							_lowercase			:     Any      					=				self.proj_in(lowerCamelCase_    )
					elif self.is_input_vectorized:
						_lowercase			:     Optional[Any]      					=				self.latent_image_embedding(lowerCamelCase_    )
					elif self.is_input_patches:
						_lowercase			:     Union[str, Any]      					=				self.pos_embed(lowerCamelCase_    )

					# 2. Blocks
					for block in self.transformer_blocks:
						_lowercase			:     Tuple      					=				block(
						    lowerCamelCase_      ,       attention_mask=lowerCamelCase_      ,       encoder_hidden_states=lowerCamelCase_      ,       encoder_attention_mask=lowerCamelCase_      ,       timestep=lowerCamelCase_      ,       cross_attention_kwargs=lowerCamelCase_      ,       class_labels=lowerCamelCase_      ,       )

					# 3. Output
					if self.is_input_continuous:
						if not self.use_linear_projection:
							_lowercase			:     Union[str, Any]      					=				hidden_states.reshape(lowerCamelCase_      ,       lowerCamelCase_      ,       lowerCamelCase_      ,       lowerCamelCase_    ).permute(0      ,       3      ,       1      ,       2    ).contiguous()
							_lowercase			:     Tuple      					=				self.proj_out(lowerCamelCase_    )
						else:
							_lowercase			:     int      					=				self.proj_out(lowerCamelCase_    )
							_lowercase			:     List[Any]      					=				hidden_states.reshape(lowerCamelCase_      ,       lowerCamelCase_      ,       lowerCamelCase_      ,       lowerCamelCase_    ).permute(0      ,       3      ,       1      ,       2    ).contiguous()

						_lowercase			:     List[str]      					=				hidden_states + residual
					elif self.is_input_vectorized:
						_lowercase			:     Dict      					=				self.norm_out(lowerCamelCase_    )
						_lowercase			:     Dict      					=				self.out(lowerCamelCase_    )
						# (batch, self.num_vector_embeds - 1, self.num_latent_pixels)
						_lowercase			:     Dict      					=				logits.permute(0      ,       2      ,       1    )

						# log(p(x_0))
						_lowercase			:     int      					=				F.log_softmax(logits.double()      ,       dim=1    ).float()
					elif self.is_input_patches:
						# TODO: cleanup!
						_lowercase			:     Tuple      					=				self.transformer_blocks[0].norma.emb(
						    lowerCamelCase_      ,       lowerCamelCase_      ,       hidden_dtype=hidden_states.dtype    )
						_lowercase					,   _lowercase			:     List[Any]      					=				self.proj_out_a(F.silu(lowerCamelCase_    )    ).chunk(2      ,       dim=1    )
						_lowercase			:     Optional[int]      					=				self.norm_out(lowerCamelCase_    ) * (1 + scale[:, None]) + shift[:, None]
						_lowercase			:     Optional[Any]      					=				self.proj_out_a(lowerCamelCase_    )

						# unpatchify
						_lowercase			:     Union[str, Any]      					=				int(hidden_states.shape[1] ** 0.5    )
						_lowercase			:     int      					=				hidden_states.reshape(
						    shape=(-1, height, width, self.patch_size, self.patch_size, self.out_channels)    )
						_lowercase			:     str      					=				torch.einsum('nhwpqc->nchpwq'      ,       lowerCamelCase_    )
						_lowercase			:     List[str]      					=				hidden_states.reshape(
						    shape=(-1, self.out_channels, height * self.patch_size, width * self.patch_size)    )

					if not return_dict:
						return (output,)

					return TransformeraDModelOutput(sample=lowerCamelCase_    )



