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