



import numpy as np
from nltk.translate import meteor_score

import datasets
from datasets.config import importlib_metadata, version


_lowercase: Dict      			=		version.parse(importlib_metadata.version("nltk"))
if NLTK_VERSION >= version.Version("3.6.4"):
				from nltk import word_tokenize


_lowercase: Any      			=		'\\n@inproceedings{banarjee2005,\n  title     = {{METEOR}: An Automatic Metric for {MT} Evaluation with Improved Correlation with Human Judgments},\n  author    = {Banerjee, Satanjeev  and Lavie, Alon},\n  booktitle = {Proceedings of the {ACL} Workshop on Intrinsic and Extrinsic Evaluation Measures for Machine Translation and/or Summarization},\n  month     = jun,\n  year      = {2005},\n  address   = {Ann Arbor, Michigan},\n  publisher = {Association for Computational Linguistics},\n  url       = {https://www.aclweb.org/anthology/W05-0909},\n  pages     = {65--72},\n}\n'

_lowercase: Optional[Any]      			=		'\\nMETEOR, an automatic metric for machine translation evaluation\nthat is based on a generalized concept of unigram matching between the\nmachine-produced translation and human-produced reference translations.\nUnigrams can be matched based on their surface forms, stemmed forms,\nand meanings; furthermore, METEOR can be easily extended to include more\nadvanced matching strategies. Once all generalized unigram matches\nbetween the two strings have been found, METEOR computes a score for\nthis matching using a combination of unigram-precision, unigram-recall, and\na measure of fragmentation that is designed to directly capture how\nwell-ordered the matched words in the machine translation are in relation\nto the reference.\n\nMETEOR gets an R correlation value of 0.347 with human evaluation on the Arabic\ndata and 0.331 on the Chinese data. This is shown to be an improvement on\nusing simply unigram-precision, unigram-recall and their harmonic F1\ncombination.\n'

_lowercase: Tuple      			=		'\nComputes METEOR score of translated segments against one or more references.\nArgs:\n    predictions: list of predictions to score. Each prediction\n        should be a string with tokens separated by spaces.\n    references: list of reference for each prediction. Each\n        reference should be a string with tokens separated by spaces.\n    alpha: Parameter for controlling relative weights of precision and recall. default: 0.9\n    beta: Parameter for controlling shape of penalty as a function of fragmentation. default: 3\n    gamma: Relative weight assigned to fragmentation penalty. default: 0.5\nReturns:\n    \'meteor\': meteor score.\nExamples:\n\n    >>> meteor = datasets.load_metric(\'meteor\')\n    >>> predictions = ["It is a guide to action which ensures that the military always obeys the commands of the party"]\n    >>> references = ["It is a guide to action that ensures that the military will forever heed Party commands"]\n    >>> results = meteor.compute(predictions=predictions, references=references)\n    >>> print(round(results["meteor"], 4))\n    0.6944\n'



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





	"""simple docstring"""




	def      UpperCamelCase_       (self		):
						"""simple docstring"""

						return datasets.MetricInfo(
						    description=_DESCRIPTION		,       citation=_CITATION		,       inputs_description=_KWARGS_DESCRIPTION		,       features=datasets.Features(
						        {
						            "predictions": datasets.Value("string"		,       id="sequence"		),
						            "references": datasets.Value("string"		,       id="sequence"		),
						        }		)		,       codebase_urls=["https://github.com/nltk/nltk/blob/develop/nltk/translate/meteor_score.py"]		,       reference_urls=[
						        "https://www.nltk.org/api/nltk.translate.html#module-nltk.translate.meteor_score",
						        "https://en.wikipedia.org/wiki/METEOR",
						    ]		,       )




	def      UpperCamelCase_       (self		,       lowerCamelCase_		):
						"""simple docstring"""

						import nltk

						nltk.download("wordnet"		)
						if NLTK_VERSION >= version.Version("3.6.5"		):
											nltk.download("punkt"		)
						if NLTK_VERSION >= version.Version("3.6.6"		):
											nltk.download("omw-1.4"		)




	def      UpperCamelCase_       (self		,       lowerCamelCase_		,       lowerCamelCase_		,       lowerCamelCase_=0.9		,       lowerCamelCase_=3		,       lowerCamelCase_=0.5		):
						"""simple docstring"""

						if NLTK_VERSION >= version.Version("3.6.5"		):
											a					     =   [
											    meteor_score.single_meteor_score(
											        word_tokenize(__a		)		,       word_tokenize(__a		)		,       alpha=__a		,       beta=__a		,       gamma=__a		)
											    for ref, pred in zip(__a		,       __a		)
											]
						else:
											a					     =   [
											    meteor_score.single_meteor_score(__a		,       __a		,       alpha=__a		,       beta=__a		,       gamma=__a		)
											    for ref, pred in zip(__a		,       __a		)
											]

						return {"meteor": np.mean(__a		)}







'''simple docstring'''







import warnings
from typing import List

import numpy as np

from ...processing_utils import ProcessorMixin
from ...tokenization_utils_base import BatchEncoding
from ...utils import is_flax_available, is_tf_available, is_torch_available
class 				__SCREAMING_SNAKE_CASE  (lowerCamelCase_      ):






				"""simple docstring"""







				__a            =['image_processor', 'tokenizer']
				__a            ='OwlViTImageProcessor'
				__a            =('CLIPTokenizer', 'CLIPTokenizerFast')




				def __init__(  self     :					List[Any]					,				__a     :					str=None					,				__a     :					List[str]=None					,				**__a     :					List[Any]					):
								_a								= None
								if "feature_extractor" in kwargs:
												warnings.warn(
												    "The `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor`"
												    " instead."					,				__a					,				)
												_a								= kwargs.pop("feature_extractor"					)

								_a								= image_processor if image_processor is not None else feature_extractor
								if image_processor is None:
												raise ValueError("You need to specify an `image_processor`."					)
								if tokenizer is None:
												raise ValueError("You need to specify a `tokenizer`."					)

								super().__init__(__a					,				__a					)




				def __call__(  self     :					Union[str, Any]					,				__a     :					Any=None					,				__a     :					List[str]=None					,				__a     :					int=None					,				__a     :					Optional[int]="max_length"					,				__a     :					List[str]="np"					,				**__a     :					Any					):

								if text is None and query_images is None and images is None:
												raise ValueError(
												    "You have to specify at least one text or query image or image. All three cannot be none."					)

								if text is not None:
												if isinstance(__a					,				__a					) or (isinstance(__a					,				__a					) and not isinstance(text[0]					,				__a					)):
																_a								= [self.tokenizer(__a					,				padding=__a					,				return_tensors=__a					,				**__a					)]

												elif isinstance(__a					,				__a					) and isinstance(text[0]					,				__a					):
																_a								= []

																# Maximum number of queries across batch
																_a								= max([len(__a					) for t in text]					)

																# Pad all batch samples to max number of text queries
																for t in text:
																				if len(__a					) != max_num_queries:
																								_a								= t + [" "] * (max_num_queries - len(__a					))

																				_a								= self.tokenizer(__a					,				padding=__a					,				return_tensors=__a					,				**__a					)
																				encodings.append(__a					)
												else:
																raise TypeError("Input text should be a string, a list of strings or a nested list of strings"					)

												if return_tensors == "np":
																_a								= np.concatenate([encoding["input_ids"] for encoding in encodings]					,				axis=0					)
																_a								= np.concatenate([encoding["attention_mask"] for encoding in encodings]					,				axis=0					)

												elif return_tensors == "jax" and is_flax_available():
																import jax.numpy as jnp

																_a								= jnp.concatenate([encoding["input_ids"] for encoding in encodings]					,				axis=0					)
																_a								= jnp.concatenate([encoding["attention_mask"] for encoding in encodings]					,				axis=0					)

												elif return_tensors == "pt" and is_torch_available():
																import torch

																_a								= torch.cat([encoding["input_ids"] for encoding in encodings]					,				dim=0					)
																_a								= torch.cat([encoding["attention_mask"] for encoding in encodings]					,				dim=0					)

												elif return_tensors == "tf" and is_tf_available():
																import tensorflow as tf

																_a								= tf.stack([encoding["input_ids"] for encoding in encodings]					,				axis=0					)
																_a								= tf.stack([encoding["attention_mask"] for encoding in encodings]					,				axis=0					)

												else:
																raise ValueError("Target return tensor type could not be returned"					)

												_a								= BatchEncoding()
												_a								= input_ids
												_a								= attention_mask

								if query_images is not None:
												_a								= BatchEncoding()
												_a								= self.image_processor(
												    __a					,				return_tensors=__a					,				**__a					).pixel_values
												_a								= query_pixel_values

								if images is not None:
												_a								= self.image_processor(__a					,				return_tensors=__a					,				**__a					)

								if text is not None and images is not None:
												_a								= image_features.pixel_values
												return encoding
								elif query_images is not None and images is not None:
												_a								= image_features.pixel_values
												return encoding
								elif text is not None or query_images is not None:
												return encoding
								else:
												return BatchEncoding(data=dict(**__a					)					,				tensor_type=__a					)




				def   UpperCamelCase__   (  self     :					List[str]					,				*__a     :					Union[str, Any]					,				**__a     :					int					):
								return self.image_processor.post_process(*__a					,				**__a					)




				def   UpperCamelCase__   (  self     :					Optional[int]					,				*__a     :					Optional[Any]					,				**__a     :					List[str]					):
								return self.image_processor.post_process_object_detection(*__a					,				**__a					)




				def   UpperCamelCase__   (  self     :					Optional[Any]					,				*__a     :					Dict					,				**__a     :					Union[str, Any]					):
								return self.image_processor.post_process_image_guided_detection(*__a					,				**__a					)




				def   UpperCamelCase__   (  self     :					str					,				*__a     :					Tuple					,				**__a     :					Tuple					):
								return self.tokenizer.batch_decode(*__a					,				**__a					)




				def   UpperCamelCase__   (  self     :					List[str]					,				*__a     :					List[Any]					,				**__a     :					Optional[int]					):
								return self.tokenizer.decode(*__a					,				**__a					)




				@property
				def   UpperCamelCase__   (  self     :					List[str]					):
								warnings.warn(
								    "`feature_extractor_class` is deprecated and will be removed in v5. Use `image_processor_class` instead."					,				__a					,				)
								return self.image_processor_class




				@property
				def   UpperCamelCase__   (  self     :					str					):
								warnings.warn(
								    "`feature_extractor` is deprecated and will be removed in v5. Use `image_processor` instead."					,				__a					,				)
								return self.image_processor
