spring_2016_project/check.py at master · XJIE6/spring_2016_project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
counter1 = 0
counter2 = 0
counter3 = 0
for (c1, a1, b1) in arr :
    for (c2, a2, b2) in arr:
        c1_vec = vectorizer1.transform([c1])[0]
        c2_vec = vectorizer1.transform([c2])[0]
        a1_full = [a for a in a1]
        if len(a1_full) == 0 :
            a1_full = [' ']

        a2_full = [a for a in a2]
        if len(a2_full) == 0 :
            a2_full = [' ']

        b1_full = [b for b in b1]
        if len(b1_full) == 0 :
            b1_full = [' ']

        b2_full = [b for b in b2]
        if len(b2_full) == 0 :
            b2_full = [' ']

        a1_vec = vectorizer2.transform(a1_full)[0]
        a2_vec = vectorizer2.transform(a2_full)[0]

        b1_vec = vectorizer3.transform(b1_full)[0]
        b2_vec = vectorizer3.transform(b2_full)[0]

        c1_label = km1.predict(c1_vec)[0]
        c2_label = km1.predict(c2_vec)[0]

        a1_label = km2.predict(a1_vec)[0]
        a2_label = km2.predict(a2_vec)[0]

        b1_label = km3.predict(b1_vec)[0]
        b2_label = km3.predict(b2_vec)[0]

        counter1 += 3
        if (c1_label == c2_label) == (a1_label == a2_label) :
            counter2 += 1

        if (c1_label == c2_label) == (b1_label == b2_label) :
            counter2 += 1

        if (b1_label == b2_label) == (a1_label == a2_label) :
            counter2 += 1

        if ((c1_label == c2_label) == (a1_label == a2_label)) == ((b1_label == b2_label) == (c1_label == c2_label)) :
            counter3 += 3

print(counter1)
print(counter2)
print(counter3)