Csar Fdez 1 month ago
parent
commit
370dd133a7
1 changed files with 17 additions and 3 deletions
  1. 17
    3
      v2_unsupervised.py

+ 17
- 3
v2_unsupervised.py View File

248
     km.to_pickle(modelpath)
248
     km.to_pickle(modelpath)
249
 else:
249
 else:
250
     km.from_pickle(modelpath)
250
     km.from_pickle(modelpath)
251
-    #km.fit_predict(xtrain)
252
 
251
 
253
 colorline=['violet','lightcoral','cyan','lime','grey']
252
 colorline=['violet','lightcoral','cyan','lime','grey']
254
 colordot=['darkviolet','red','blue','green','black']
253
 colordot=['darkviolet','red','blue','green','black']
379
     TP=[]
378
     TP=[]
380
     FN=[]
379
     FN=[]
381
     TPFP=[]
380
     TPFP=[]
381
+    A=[]
382
+    B=[]
383
+    COUNTER=[]
384
+    ASSIGNED=[]
382
     for i in range(NumberOfFailures+1):
385
     for i in range(NumberOfFailures+1):
386
+        A.append(i)
383
         TP.append([])
387
         TP.append([])
384
         FN.append([])
388
         FN.append([])
385
         TPFP.append([])
389
         TPFP.append([])
386
-        b=Counter(labels[ranges[i][0]:ranges[i][1]])
387
-        lab.append(b.most_common(1)[0][0])
390
+        COUNTER.append(Counter(labels[ranges[i][0]:ranges[i][1]]))
391
+
392
+    while len(A)>0:
393
+        a=A.pop(0)
394
+        Class=COUNTER[a].most_common(1)[0][0]
395
+        if Class in ASSIGNED:
396
+            B.append(Class)
397
+        else:
398
+            ASSIGNED.append(Class)       
399
+
400
+    # AKI
401
+    #lab.append(b.most_common(1)[0][0])
388
 
402
 
389
     for i in range(NumberOfFailures+1):
403
     for i in range(NumberOfFailures+1):
390
         counttp=0
404
         counttp=0

Powered by TurnKey Linux.