Spaces:
Running
Running
Update Space (evaluate main: c447fc8e)
Browse files- mahalanobis.py +2 -22
- requirements.txt +1 -1
mahalanobis.py
CHANGED
@@ -13,9 +13,6 @@
|
|
13 |
# limitations under the License.
|
14 |
"""Mahalanobis metric."""
|
15 |
|
16 |
-
from dataclasses import dataclass
|
17 |
-
from typing import List, Optional
|
18 |
-
|
19 |
import datasets
|
20 |
import numpy as np
|
21 |
|
@@ -60,25 +57,13 @@ Examples:
|
|
60 |
"""
|
61 |
|
62 |
|
63 |
-
@dataclass
|
64 |
-
class MahalanobisConfig(evaluate.info.Config):
|
65 |
-
|
66 |
-
name: str = "default"
|
67 |
-
|
68 |
-
reference_distribution: Optional[List] = None
|
69 |
-
|
70 |
-
|
71 |
@evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
|
72 |
class Mahalanobis(evaluate.Metric):
|
73 |
-
|
74 |
-
ALLOWED_CONFIG_NAMES = ["default"]
|
75 |
-
|
76 |
-
def _info(self, config):
|
77 |
return evaluate.MetricInfo(
|
78 |
description=_DESCRIPTION,
|
79 |
citation=_CITATION,
|
80 |
inputs_description=_KWARGS_DESCRIPTION,
|
81 |
-
config=config,
|
82 |
features=datasets.Features(
|
83 |
{
|
84 |
"X": datasets.Sequence(datasets.Value("float", id="sequence"), id="X"),
|
@@ -86,12 +71,7 @@ class Mahalanobis(evaluate.Metric):
|
|
86 |
),
|
87 |
)
|
88 |
|
89 |
-
def _compute(self, X):
|
90 |
-
|
91 |
-
if self.config.reference_distribution is None:
|
92 |
-
raise ValueError("You need to provide a `reference_distribution`.")
|
93 |
-
else:
|
94 |
-
reference_distribution = self.config.reference_distribution
|
95 |
|
96 |
# convert to numpy arrays
|
97 |
X = np.array(X)
|
|
|
13 |
# limitations under the License.
|
14 |
"""Mahalanobis metric."""
|
15 |
|
|
|
|
|
|
|
16 |
import datasets
|
17 |
import numpy as np
|
18 |
|
|
|
57 |
"""
|
58 |
|
59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
@evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
|
61 |
class Mahalanobis(evaluate.Metric):
|
62 |
+
def _info(self):
|
|
|
|
|
|
|
63 |
return evaluate.MetricInfo(
|
64 |
description=_DESCRIPTION,
|
65 |
citation=_CITATION,
|
66 |
inputs_description=_KWARGS_DESCRIPTION,
|
|
|
67 |
features=datasets.Features(
|
68 |
{
|
69 |
"X": datasets.Sequence(datasets.Value("float", id="sequence"), id="X"),
|
|
|
71 |
),
|
72 |
)
|
73 |
|
74 |
+
def _compute(self, X, reference_distribution):
|
|
|
|
|
|
|
|
|
|
|
75 |
|
76 |
# convert to numpy arrays
|
77 |
X = np.array(X)
|
requirements.txt
CHANGED
@@ -1 +1 @@
|
|
1 |
-
git+https://github.com/huggingface/evaluate@
|
|
|
1 |
+
git+https://github.com/huggingface/evaluate@c447fc8eda9c62af501bfdc6988919571050d950
|