"""simple docstring"""


from sklearn.metrics import accuracy_score

import datasets


lowerCAmelCase_:       List[Any]			     =       "\nAccuracy is the proportion of correct predictions among the total number of cases processed. It can be computed with:\nAccuracy = (TP + TN) / (TP + TN + FP + FN)\n Where:\nTP: True positive\nTN: True negative\nFP: False positive\nFN: False negative\n"


lowerCAmelCase_:       Union[str, Any]			     =       "\nArgs:\n    predictions (`list` of `int`): Predicted labels.\n    references (`list` of `int`): Ground truth labels.\n    normalize (`boolean`): If set to False, returns the number of correctly classified samples. Otherwise, returns the fraction of correctly classified samples. Defaults to True.\n    sample_weight (`list` of `float`): Sample weights Defaults to None.\n\nReturns:\n    accuracy (`float` or `int`): Accuracy score. Minimum possible value is 0. Maximum possible value is 1.0, or the number of examples input, if `normalize` is set to `True`.. A higher score means higher accuracy.\n\nExamples:\n\n    Example 1-A simple example\n        >>> accuracy_metric = datasets.load_metric(\"accuracy\")\n        >>> results = accuracy_metric.compute(references=[0, 1, 2, 0, 1, 2], predictions=[0, 1, 1, 2, 1, 0])\n        >>> print(results)\n        {'accuracy': 0.5}\n\n    Example 2-The same as Example 1, except with `normalize` set to `False`.\n        >>> accuracy_metric = datasets.load_metric(\"accuracy\")\n        >>> results = accuracy_metric.compute(references=[0, 1, 2, 0, 1, 2], predictions=[0, 1, 1, 2, 1, 0], normalize=False)\n        >>> print(results)\n        {'accuracy': 3.0}\n\n    Example 3-The same as Example 1, except with `sample_weight` set.\n        >>> accuracy_metric = datasets.load_metric(\"accuracy\")\n        >>> results = accuracy_metric.compute(references=[0, 1, 2, 0, 1, 2], predictions=[0, 1, 1, 2, 1, 0], sample_weight=[0.5, 2, 0.7, 0.5, 9, 0.4])\n        >>> print(results)\n        {'accuracy': 0.8778625954198473}\n"


lowerCAmelCase_:       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  a__     ( datasets.Metric						):


   def 						snake_case__						(  self	):
        '''simple docstring'''



        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.accuracy_score.html"],					)


   def 						snake_case__						(  self,					_UpperCAmelCase,					_UpperCAmelCase,					_UpperCAmelCase=True,					_UpperCAmelCase=None	):
        '''simple docstring'''



        return {
            "accuracy": float(
                accuracy_score(A_,					A_,					normalize=A_,					sample_weight=A_	)	)
        }







from typing import SupportsIndex

import numpy as np
from scipy.ndimage import convolve





def 				__a							(       __A			,							__A			,							__A						):
    SCREAMING_SNAKE_CASE__         =    np.stack((horizontal_flow, vertical_flow)			,							2						)

    # Create a grid of all pixel coordinates and subtract the flow to get the
    # target pixels coordinates
    SCREAMING_SNAKE_CASE__         =    np.stack(
        np.meshgrid(np.arange(0			,							image.shape[1]						)			,							np.arange(0			,							image.shape[0]						)						)			,							2						)
    SCREAMING_SNAKE_CASE__         =    np.round(grid - flow						).astype(np.intaa						)

    # Find the locations outside of the original image
    SCREAMING_SNAKE_CASE__         =    (grid < 0) | (grid >= np.array([image.shape[1], image.shape[0]]						))
    SCREAMING_SNAKE_CASE__         =    0

    SCREAMING_SNAKE_CASE__         =    image[grid[:, :, 1], grid[:, :, 0]]

    # Set pixels at invalid locations to 0
    SCREAMING_SNAKE_CASE__         =    0

    return warped





def 				__a							(       __A			,							__A			,							__A			,							__A = None			,							):
    if alpha is None:
        SCREAMING_SNAKE_CASE__         =    0.1

    # Initialize flow
    SCREAMING_SNAKE_CASE__         =    np.zeros_like(__A						)
    SCREAMING_SNAKE_CASE__         =    np.zeros_like(__A						)

    # Prepare kernels for the calculation of the derivatives and the average velocity
    SCREAMING_SNAKE_CASE__         =    np.array([[-1, 1], [-1, 1]]						) * 0.25
    SCREAMING_SNAKE_CASE__         =    np.array([[-1, -1], [1, 1]]						) * 0.25
    SCREAMING_SNAKE_CASE__         =    np.array([[1, 1], [1, 1]]						) * 0.25
    SCREAMING_SNAKE_CASE__         =    np.array(
        [[1 / 12, 1 / 6, 1 / 12], [1 / 6, 0, 1 / 6], [1 / 12, 1 / 6, 1 / 12]]						)

    # Iteratively refine the flow
    for _ in range(__A						):
        SCREAMING_SNAKE_CASE__         =    warp(__A			,							__A			,							__A						)
        SCREAMING_SNAKE_CASE__         =    convolve(__A			,							__A						) + convolve(__A			,							__A						)
        SCREAMING_SNAKE_CASE__         =    convolve(__A			,							__A						) + convolve(__A			,							__A						)
        SCREAMING_SNAKE_CASE__         =    convolve(__A			,							__A						) + convolve(__A			,							-kernel_t						)

        SCREAMING_SNAKE_CASE__         =    convolve(__A			,							__A						)
        SCREAMING_SNAKE_CASE__         =    convolve(__A			,							__A						)

        # This updates the flow as proposed in the paper (Step 12)
        SCREAMING_SNAKE_CASE__         =    (
            derivative_x * avg_horizontal_velocity
            + derivative_y * avg_vertical_velocity
            + derivative_t
        )
        SCREAMING_SNAKE_CASE__         =    update / (alpha**2 + derivative_x**2 + derivative_y**2)

        SCREAMING_SNAKE_CASE__         =    avg_horizontal_velocity - derivative_x * update
        SCREAMING_SNAKE_CASE__         =    avg_vertical_velocity - derivative_y * update

    return horizontal_flow, vertical_flow


if __name__ == "__main__":
  import doctest

  doctest.testmod()


