.

v2_unsupervised.py

@@ -248,7 +248,6 @@ if options.train:
     km.to_pickle(modelpath)
 else:
     km.from_pickle(modelpath)
-    #km.fit_predict(xtrain)
 
 colorline=['violet','lightcoral','cyan','lime','grey']
 colordot=['darkviolet','red','blue','green','black']
@@ -379,12 +378,27 @@ def anomalyMetric(labels,ranges):
     TP=[]
     FN=[]
     TPFP=[]
+    A=[]
+    B=[]
+    COUNTER=[]
+    ASSIGNED=[]
     for i in range(NumberOfFailures+1):
+        A.append(i)
         TP.append([])
         FN.append([])
         TPFP.append([])
-        b=Counter(labels[ranges[i][0]:ranges[i][1]])
-        lab.append(b.most_common(1)[0][0])
+        COUNTER.append(Counter(labels[ranges[i][0]:ranges[i][1]]))
+
+    while len(A)>0:
+        a=A.pop(0)
+        Class=COUNTER[a].most_common(1)[0][0]
+        if Class in ASSIGNED:
+            B.append(Class)
+        else:
+            ASSIGNED.append(Class)       
+
+    # AKI
+    #lab.append(b.most_common(1)[0][0])
 
     for i in range(NumberOfFailures+1):
         counttp=0