From 09543f9c7e699f74a1ba2a9418e2323803eaccab Mon Sep 17 00:00:00 2001 From: lhenry15 Date: Thu, 6 May 2021 13:26:29 -0500 Subject: [PATCH] fix LSTMODetector skitest --- tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py b/tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py index dd6359b..bfe11df 100644 --- a/tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py +++ b/tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py @@ -29,7 +29,8 @@ class LSTMODetectorSKI_TestCase(unittest.TestCase): self.y_test = self.y_test[1:] self.y_train = self.y_train[1:] - self.transformer = LSTMODetectorSKI(contamination=self.contamination) + + self.transformer = LSTMODetectorSKI(contamination=self.contamination, feature_dim=self.X_train.shape[1]) self.transformer.fit(self.X_train) def test_prediction_labels(self):