From 0c0eb8792bfe048efd2740aa63cd8ff98162057c Mon Sep 17 00:00:00 2001 From: Shiyu Wang <42013343+shi-yu-wang@users.noreply.github.com> Date: Wed, 12 Jun 2024 10:59:11 -0700 Subject: [PATCH] lower threshold --- tests/anomaly/test_dpad.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/tests/anomaly/test_dpad.py b/tests/anomaly/test_dpad.py index e91a1d60..dca62572 100644 --- a/tests/anomaly/test_dpad.py +++ b/tests/anomaly/test_dpad.py @@ -51,7 +51,7 @@ def __init__(self, *args, **kwargs): transform=TransformSequence( [TemporalResample("15min"), Shingle(size=3, stride=2), DifferenceTransform()] ), - threshold = AdaptiveAggregateAlarms(0.1) + threshold = AdaptiveAggregateAlarms(0.0001) ) ) @@ -81,12 +81,7 @@ def test_full(self): self.model.save(dirname=join(rootdir, "tmp", "dpad")) loaded_model = DeepPointAnomalyDetector.load(dirname=join(rootdir, "tmp", "dpad")) loaded_alarms = loaded_model.get_anomaly_label(self.test_data) - if sys.version_info[1] < 8: - n_loaded_alarms = sum(loaded_alarms.to_pd().values != 0) - - if sys.version_info[1] >= 8: - n_loaded_alarms = np.sum(loaded_alarms.to_pd().values != 0) - + n_loaded_alarms = sum(loaded_alarms.to_pd().values != 0) self.assertAlmostEqual(n_loaded_alarms, n_alarms, delta=1) # Evaluation @@ -100,5 +95,4 @@ def test_full(self): logging.basicConfig( format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s", stream=sys.stdout, level=logging.DEBUG ) - unittest.main() - + unittest.main() \ No newline at end of file