Browse Source

fix ski KDiscord unittest

master
lhenry15 4 years ago
parent
commit
439addcb12
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      tods/tests/sk_interface/detection_algorithm/test_ski_KDiscordODetector.py

+ 3
- 1
tods/tests/sk_interface/detection_algorithm/test_ski_KDiscordODetector.py View File

@@ -22,11 +22,13 @@ class KDiscordODetectorSKI_TestCase(unittest.TestCase):
self.n_test = 100
self.contamination = 0.1
self.roc_floor = 0.0
self.window_size = 5
self.X_train, self.y_train, self.X_test, self.y_test = generate_data(
n_train=self.n_train, n_test=self.n_test,
contamination=self.contamination, random_state=42)

self.transformer = KDiscordODetectorSKI(contamination=self.contamination)
self.transformer = KDiscordODetectorSKI(contamination=self.contamination, window_size=self.window_size)
self.y_test = self.y_test[self.window_size-1:]
self.transformer.fit(self.X_train)

def test_prediction_labels(self):


Loading…
Cancel
Save