Browse Source

fix LSTMODetector skitest

master
lhenry15 4 years ago
parent
commit
09543f9c7e
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py

+ 2
- 1
tods/tests/sk_interface/detection_algorithm/test_ski_LSTMODetector.py View File

@@ -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):


Loading…
Cancel
Save