OpenI
/
tods
Not watched
1
0
Fork 0
Code
Releases 0 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix LSTMODetector skitest

master
lhenry15 4 years ago
parent
ca43ede1c5
commit
09543f9c7e
1 changed files with 2 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  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_test = self.y_test[1:]
self.y_train = self.y_train[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) self.transformer.fit(self.X_train)


def test_prediction_labels(self): def test_prediction_labels(self):


Write Preview
Loading…
Cancel
Save