10
10
11
11
12
12
from sklearn .metrics import roc_auc_score
13
- from scipy .stats import rankdata
14
13
15
14
from pyod .utils .data import generate_data
16
15
from pyod .models .knn import KNN
@@ -30,8 +29,8 @@ def setUp(self):
30
29
self .n_test = 100
31
30
self .contamination = 0.1
32
31
self .roc_floor = 0.8
33
- self .X_train , self .y_train , self .X_test , self .y_test = generate_data (
34
- n_train = self .n_train , n_test = self .n_test ,
32
+ self .X_train , self .X_test , self .y_train , self .y_test = generate_data (
33
+ n_train = self .n_train , n_test = self .n_test , behaviour = 'new' ,
35
34
contamination = self .contamination , random_state = 42 )
36
35
37
36
detectors = [KNN (), LOF (), OCSVM ()]
@@ -98,8 +97,8 @@ def setUp(self):
98
97
self .n_test = 100
99
98
self .contamination = 0.1
100
99
self .roc_floor = 0.8
101
- self .X_train , self .y_train , self .X_test , self .y_test = generate_data (
102
- n_train = self .n_train , n_test = self .n_test ,
100
+ self .X_train , self .X_test , self .y_train , self .y_test = generate_data (
101
+ n_train = self .n_train , n_test = self .n_test , behaviour = 'new' ,
103
102
contamination = self .contamination , random_state = 42 )
104
103
105
104
detectors = [KNN (), LOF (), OCSVM ()]
@@ -166,8 +165,8 @@ def setUp(self):
166
165
self .n_test = 100
167
166
self .contamination = 0.1
168
167
self .roc_floor = 0.8
169
- self .X_train , self .y_train , self .X_test , self .y_test = generate_data (
170
- n_train = self .n_train , n_test = self .n_test ,
168
+ self .X_train , self .X_test , self .y_train , self .y_test = generate_data (
169
+ n_train = self .n_train , n_test = self .n_test , behaviour = 'new' ,
171
170
contamination = self .contamination , random_state = 42 )
172
171
173
172
detectors = [KNN (), LOF (), OCSVM ()]
0 commit comments