Spaces:
Running
Running
| from sklearn_proxy import SklearnProxy | |
| import unittest | |
| accuracy_test_cases = [ | |
| { | |
| "predictions": [0, 0], | |
| "references": [1, 1], | |
| "result": {"accuracy": 0.0} | |
| }, | |
| { | |
| "predictions": [1, 1], | |
| "references": [1, 1], | |
| "result": {"accuracy": 1.0} | |
| }, | |
| { | |
| "predictions": [1, 0], | |
| "references": [1, 1], | |
| "result": {"accuracy": 0.5} | |
| } | |
| ] | |
| class TestGeneral(unittest.TestCase): | |
| def test_accuracy(self): | |
| metric = SklearnProxy() | |
| for test_case in accuracy_test_cases: | |
| result = metric.compute(predictions=test_case["predictions"],references=test_case["references"], metric_name="accuracy") | |
| self.assertEqual(result, test_case["result"]) | |