


from sklearn.metrics import mean_squared_error

import datasets


a_ :Tuple  							=			'\\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'

a_ :Optional[Any]  							=			'\\nMean Squared Error(MSE) is the average of the square of difference between the predicted\nand actual values.\n'


a_ :int  							=			'\nArgs:\n    predictions: array-like of shape (n_samples,) or (n_samples, n_outputs)\n        Estimated target values.\n    references: array-like of shape (n_samples,) or (n_samples, n_outputs)\n        Ground truth (correct) target values.\n    sample_weight: array-like of shape (n_samples,), default=None\n        Sample weights.\n    multioutput: {"raw_values", "uniform_average"} or array-like of shape (n_outputs,), default="uniform_average"\n        Defines aggregating of multiple output values. Array-like value defines weights used to average errors.\n\n        "raw_values" : Returns a full set of errors in case of multioutput input.\n\n        "uniform_average" : Errors of all outputs are averaged with uniform weight.\n\n    squared : bool, default=True\n        If True returns MSE value, if False returns RMSE (Root Mean Squared Error) value.\n\nReturns:\n    mse : mean squared error.\nExamples:\n\n    >>> mse_metric = datasets.load_metric("mse")\n    >>> predictions = [2.5, 0.0, 2, 8]\n    >>> references = [3, -0.5, 2, 7]\n    >>> results = mse_metric.compute(predictions=predictions, references=references)\n    >>> print(results)\n    {\'mse\': 0.375}\n    >>> rmse_result = mse_metric.compute(predictions=predictions, references=references, squared=False)\n    >>> print(rmse_result)\n    {\'mse\': 0.6123724356957945}\n\n    If you\'re using multi-dimensional lists, then set the config as follows :\n\n    >>> mse_metric = datasets.load_metric("mse", "multilist")\n    >>> predictions = [[0.5, 1], [-1, 1], [7, -6]]\n    >>> references = [[0, 2], [-1, 2], [8, -5]]\n    >>> results = mse_metric.compute(predictions=predictions, references=references)\n    >>> print(results)\n    {\'mse\': 0.7083333333333334}\n    >>> results = mse_metric.compute(predictions=predictions, references=references, multioutput=\'raw_values\')\n    >>> print(results) # doctest: +NORMALIZE_WHITESPACE\n    {\'mse\': array([0.41666667, 1. ])}\n'

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







     def 							lowercase__       (						self :						Dict      ):
           return datasets.MetricInfo(
               description=_DESCRIPTION   ,   citation=_CITATION   ,   inputs_description=_KWARGS_DESCRIPTION   ,   features=datasets.Features(self._get_feature_types()      )   ,   reference_urls=[
                   '''https://scikit-learn.org/stable/modules/generated/sklearn.metrics.mean_squared_error.html'''
               ]   ,   )







     def 							lowercase__       (						self :						Dict      ):
           if self.config_name == "multilist":
                 return {
                     "predictions": datasets.Sequence(datasets.Value('''float'''      )      ),
                     "references": datasets.Sequence(datasets.Value('''float'''      )      ),
                 }
           else:
                 return {
                     "predictions": datasets.Value('''float'''      ),
                     "references": datasets.Value('''float'''      ),
                 }







     def 							lowercase__       (						self :						str   ,   _lowercase :						str   ,   _lowercase :						int   ,   _lowercase :						int=None   ,   _lowercase :						Dict="uniform_average"   ,   _lowercase :						List[str]=True      ):
           SCREAMING_SNAKE_CASE__       :  str						     =		mean_squared_error(
               _lowercase   ,   _lowercase   ,   sample_weight=_lowercase   ,   multioutput=_lowercase   ,   squared=_lowercase      )

           return {"mse": mse}













'''simple docstring'''

from datasets.utils.patching import _PatchedModuleObj, patch_submodule

from . import _test_patching







def    a__							(							)						->						Tuple:





     """simple docstring"""

     import os as original_os
     from os import path as original_path
     from os import rename as original_rename
     from os.path import dirname as original_dirname
     from os.path import join as original_join

     assert _test_patching.os is original_os
     assert _test_patching.path is original_path
     assert _test_patching.join is original_join

     assert _test_patching.renamed_os is original_os
     assert _test_patching.renamed_path is original_path
     assert _test_patching.renamed_join is original_join

     _UpperCamelCase							       =			'''__test_patch_submodule_mock__'''
     with patch_submodule(_test_patching,   '''os.path.join''',   lowercase ):
          # Every way to access os.path.join must be patched, and the rest must stay untouched

          # check os.path.join
          assert isinstance(_test_patching.os,   _PatchedModuleObj )
          assert isinstance(_test_patching.os.path,   _PatchedModuleObj )
          assert _test_patching.os.path.join is mock

          # check path.join
          assert isinstance(_test_patching.path,   _PatchedModuleObj )
          assert _test_patching.path.join is mock

          # check join
          assert _test_patching.join is mock

          # check that the other attributes are untouched
          assert _test_patching.os.rename is original_rename
          assert _test_patching.path.dirname is original_dirname
          assert _test_patching.os.path.dirname is original_dirname

          # Even renamed modules or objects must be patched

          # check renamed_os.path.join
          assert isinstance(_test_patching.renamed_os,   _PatchedModuleObj )
          assert isinstance(_test_patching.renamed_os.path,   _PatchedModuleObj )
          assert _test_patching.renamed_os.path.join is mock

          # check renamed_path.join
          assert isinstance(_test_patching.renamed_path,   _PatchedModuleObj )
          assert _test_patching.renamed_path.join is mock

          # check renamed_join
          assert _test_patching.renamed_join is mock

          # check that the other attributes are untouched
          assert _test_patching.renamed_os.rename is original_rename
          assert _test_patching.renamed_path.dirname is original_dirname
          assert _test_patching.renamed_os.path.dirname is original_dirname

     # check that everthing is back to normal when the patch is over

     assert _test_patching.os is original_os
     assert _test_patching.path is original_path
     assert _test_patching.join is original_join

     assert _test_patching.renamed_os is original_os
     assert _test_patching.renamed_path is original_path
     assert _test_patching.renamed_join is original_join







def    a__							(							)						->						Tuple:





     """simple docstring"""

     assert _test_patching.open is open

     _UpperCamelCase							       =			'''__test_patch_submodule_builtin_mock__'''
     # _test_patching has "open" in its globals
     assert _test_patching.open is open
     with patch_submodule(_test_patching,   '''open''',   lowercase ):
          assert _test_patching.open is mock

     # check that everthing is back to normal when the patch is over

     assert _test_patching.open is open







def    a__							(							)						->						Optional[Any]:





     """simple docstring"""

     _UpperCamelCase							       =			'''__test_patch_submodule_missing_mock__'''
     with patch_submodule(_test_patching,   '''pandas.read_csv''',   lowercase ):
          pass







def    a__							(							)						->						Optional[int]:





     """simple docstring"""

     _UpperCamelCase							       =			'''__test_patch_submodule_missing_builtin_mock__'''
     # _test_patching doesn't have "len" in its globals
     assert getattr(_test_patching,   '''len''',   lowercase ) is None
     with patch_submodule(_test_patching,   '''len''',   lowercase ):
          assert _test_patching.len is mock
     assert _test_patching.len is len







def    a__							(							)						->						Optional[Any]:





     """simple docstring"""

     _UpperCamelCase							       =			'''__test_patch_submodule_start_and_stop_mock__'''
     _UpperCamelCase							       =			patch_submodule(_test_patching,   '''open''',   lowercase )
     assert _test_patching.open is open
     patch.start()
     assert _test_patching.open is mock
     patch.stop()
     assert _test_patching.open is open







def    a__							(							)						->						List[str]:





     """simple docstring"""

     from os import rename as original_rename
     from os.path import dirname as original_dirname
     from os.path import join as original_join

     _UpperCamelCase							       =			'''__test_patch_submodule_successive_join__'''
     _UpperCamelCase							       =			'''__test_patch_submodule_successive_dirname__'''
     _UpperCamelCase							       =			'''__test_patch_submodule_successive_rename__'''
     assert _test_patching.os.path.join is original_join
     assert _test_patching.os.path.dirname is original_dirname
     assert _test_patching.os.rename is original_rename

     with patch_submodule(_test_patching,   '''os.path.join''',   lowercase ):
          with patch_submodule(_test_patching,   '''os.rename''',   lowercase ):
               with patch_submodule(_test_patching,   '''os.path.dirname''',   lowercase ):
                    assert _test_patching.os.path.join is mock_join
                    assert _test_patching.os.path.dirname is mock_dirname
                    assert _test_patching.os.rename is mock_rename

    # try another order
     with patch_submodule(_test_patching,   '''os.rename''',   lowercase ):
          with patch_submodule(_test_patching,   '''os.path.join''',   lowercase ):
               with patch_submodule(_test_patching,   '''os.path.dirname''',   lowercase ):
                    assert _test_patching.os.path.join is mock_join
                    assert _test_patching.os.path.dirname is mock_dirname
                    assert _test_patching.os.rename is mock_rename

     assert _test_patching.os.path.join is original_join
     assert _test_patching.os.path.dirname is original_dirname
     assert _test_patching.os.rename is original_rename







def    a__							(							)						->						str:





     """simple docstring"""

     _UpperCamelCase							       =			'''__test_patch_submodule_doesnt_exist_mock__'''
     with patch_submodule(_test_patching,   '''__module_that_doesn_exist__.__attribute_that_doesn_exist__''',   lowercase ):
          pass
     with patch_submodule(_test_patching,   '''os.__attribute_that_doesn_exist__''',   lowercase ):
          pass



