


"""simple docstring"""


import datasets

from .evaluate import evaluate


__lowerCamelCase      :int	    =		'\\n@inproceedings{Rajpurkar2016SQuAD10,\n  title={SQuAD: 100, 000+ Questions for Machine Comprehension of Text},\n  author={Pranav Rajpurkar and Jian Zhang and Konstantin Lopyrev and Percy Liang},\n  booktitle={EMNLP},\n  year={2016}\n}\n'

__lowerCamelCase      :Optional[Any]	    =		'\nThis metric wrap the official scoring script for version 1 of the Stanford Question Answering Dataset (SQuAD).\n\nStanford Question Answering Dataset (SQuAD) is a reading comprehension dataset, consisting of questions posed by\ncrowdworkers on a set of Wikipedia articles, where the answer to every question is a segment of text, or span,\nfrom the corresponding reading passage, or the question might be unanswerable.\n'

__lowerCamelCase      :str	    =		'\nComputes SQuAD scores (F1 and EM).\nArgs:\n    predictions: List of question-answers dictionaries with the following key-values:\n        - \'id\': id of the question-answer pair as given in the references (see below)\n        - \'prediction_text\': the text of the answer\n    references: List of question-answers dictionaries with the following key-values:\n        - \'id\': id of the question-answer pair (see above),\n        - \'answers\': a Dict in the SQuAD dataset format\n            {\n                \'text\': list of possible texts for the answer, as a list of strings\n                \'answer_start\': list of start positions for the answer, as a list of ints\n            }\n            Note that answer_start values are not taken into account to compute the metric.\nReturns:\n    \'exact_match\': Exact match (the normalized answer exactly match the gold answer)\n    \'f1\': The F-score of predicted tokens versus the gold answer\nExamples:\n\n    >>> predictions = [{\'prediction_text\': \'1976\', \'id\': \'56e10a3be3433e1400422b22\'}]\n    >>> references = [{\'answers\': {\'answer_start\': [97], \'text\': [\'1976\']}, \'id\': \'56e10a3be3433e1400422b22\'}]\n    >>> squad_metric = datasets.load_metric("squad")\n    >>> results = squad_metric.compute(predictions=predictions, references=references)\n    >>> print(results)\n    {\'exact_match\': 100.0, \'f1\': 100.0}\n'


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


     """simple docstring"""

     def 		a__						(	self:						str	)->    List[str]:
           return datasets.MetricInfo(
               description=_DESCRIPTION     ,		citation=_CITATION     ,		inputs_description=_KWARGS_DESCRIPTION     ,		features=datasets.Features(
                   {
                       """predictions""": {"""id""": datasets.Value("""string"""	), """prediction_text""": datasets.Value("""string"""	)},
                       """references""": {
                           """id""": datasets.Value("""string"""	),
                           """answers""": datasets.features.Sequence(
                               {
                                   """text""": datasets.Value("""string"""	),
                                   """answer_start""": datasets.Value("""int32"""	),
                               }	),
                       },
                   }	)     ,		codebase_urls=["""https://rajpurkar.github.io/SQuAD-explorer/"""]     ,		reference_urls=["""https://rajpurkar.github.io/SQuAD-explorer/"""]     ,		)







     def 		a__						(	self:						Dict     ,		__a:						Optional[Any]     ,		__a:						List[str]	)->    Tuple:
           lowerCamelCase					:					str        =	{prediction['''id''']: prediction['''prediction_text'''] for prediction in predictions}
           lowerCamelCase					:					int        =	[
               {
                   '''paragraphs''': [
                       {
                           '''qas''': [
                               {
                                   '''answers''': [{'''text''': answer_text} for answer_text in ref['''answers''']['''text''']],
                                   '''id''': ref['''id'''],
                               }
                               for ref in references
                           ]
                       }
                   ]
               }
           ]
           lowerCamelCase					:					Union[str, Any]        =	evaluate(dataset=__a     ,		predictions=__a	)
           return score








"""simple docstring"""


from typing import List, Optional, Union

from ...configuration_utils import PretrainedConfig
from ...utils import logging


__lowerCamelCase      :str	    =		logging.get_logger(__name__)

__lowerCamelCase      :Any	    =		{
    'huggingface/time-series-transformer-tourism-monthly': (
        'https://huggingface.co/huggingface/time-series-transformer-tourism-monthly/resolve/main/config.json'
    ),
    # See all TimeSeriesTransformer models at https://huggingface.co/models?filter=time_series_transformer
}


class 			A__	(						__lowercase):


     """simple docstring"""
     snake_case__	:					List[Any]													='''time_series_transformer'''
     snake_case__	:					List[Any]													={
         '''hidden_size''': '''d_model''',
         '''num_attention_heads''': '''encoder_attention_heads''',
         '''num_hidden_layers''': '''encoder_layers''',
     }

     def __init__(	self:						List[str]     ,		__a:						Optional[int] = None     ,		__a:						Optional[int] = None     ,		__a:						str = "student_t"     ,		__a:						str = "nll"     ,		__a:						int = 1     ,		__a:						List[int] = [1, 2, 3, 4, 5, 6, 7]     ,		__a:						Optional[Union[str, bool]] = "mean"     ,		__a:						int = 0     ,		__a:						int = 0     ,		__a:						int = 0     ,		__a:						int = 0     ,		__a:						Optional[List[int]] = None     ,		__a:						Optional[List[int]] = None     ,		__a:						int = 32     ,		__a:						int = 32     ,		__a:						int = 2     ,		__a:						int = 2     ,		__a:						int = 2     ,		__a:						int = 2     ,		__a:						bool = True     ,		__a:						str = "gelu"     ,		__a:						int = 64     ,		__a:						float = 0.1     ,		__a:						float = 0.1     ,		__a:						float = 0.1     ,		__a:						float = 0.1     ,		__a:						float = 0.1     ,		__a:						int = 100     ,		__a:						float = 0.02     ,		__a:						Tuple=True     ,		**__a:						str     ,		)->    Any:
           # time series specific configuration
           lowerCamelCase					:					str        =	prediction_length
           lowerCamelCase					:					Optional[Any]        =	context_length or prediction_length
           lowerCamelCase					:					Tuple        =	distribution_output
           lowerCamelCase					:					Any        =	loss
           lowerCamelCase					:					List[Any]        =	input_size
           lowerCamelCase					:					int        =	num_time_features
           lowerCamelCase					:					Dict        =	lags_sequence
           lowerCamelCase					:					Optional[int]        =	scaling
           lowerCamelCase					:					int        =	num_dynamic_real_features
           lowerCamelCase					:					Tuple        =	num_static_real_features
           lowerCamelCase					:					Any        =	num_static_categorical_features
           if cardinality and num_static_categorical_features > 0:
                 if len(__a	) != num_static_categorical_features:
                       raise ValueError(
                           """The cardinality should be a list of the same length as `num_static_categorical_features`"""	)
                 lowerCamelCase					:					int        =	cardinality
           else:
                 lowerCamelCase					:					Dict        =	[0]
           if embedding_dimension and num_static_categorical_features > 0:
                 if len(__a	) != num_static_categorical_features:
                       raise ValueError(
                           """The embedding dimension should be a list of the same length as `num_static_categorical_features`"""	)
                 lowerCamelCase					:					str        =	embedding_dimension
           else:
                 lowerCamelCase					:					str        =	[min(50     ,		(cat + 1) // 2	) for cat in self.cardinality]
           lowerCamelCase					:					Any        =	num_parallel_samples

           # Transformer architecture configuration
           lowerCamelCase					:					Any        =	input_size * len(__a	) + self._number_of_features
           lowerCamelCase					:					List[str]        =	d_model
           lowerCamelCase					:					Tuple        =	encoder_attention_heads
           lowerCamelCase					:					Optional[int]        =	decoder_attention_heads
           lowerCamelCase					:					Union[str, Any]        =	encoder_ffn_dim
           lowerCamelCase					:					str        =	decoder_ffn_dim
           lowerCamelCase					:					str        =	encoder_layers
           lowerCamelCase					:					Any        =	decoder_layers

           lowerCamelCase					:					Optional[int]        =	dropout
           lowerCamelCase					:					List[str]        =	attention_dropout
           lowerCamelCase					:					Tuple        =	activation_dropout
           lowerCamelCase					:					Optional[int]        =	encoder_layerdrop
           lowerCamelCase					:					int        =	decoder_layerdrop

           lowerCamelCase					:					Optional[int]        =	activation_function
           lowerCamelCase					:					Optional[Any]        =	init_std

           lowerCamelCase					:					Optional[Any]        =	use_cache

           super().__init__(is_encoder_decoder=__a     ,		**__a	)







     @property
     def 		a__						(	self:						int	)->    int:
           return (
               sum(self.embedding_dimension	)
               + self.num_dynamic_real_features
               + self.num_time_features
               + self.num_static_real_features
               + self.input_size * 2  # the log1p(abs(loc)) and log(scale) features
           )




