In [1]:

    

from os import listdir
from os.path import isfile, join
from sklearn.metrics import normalized_mutual_info_score


    

    




---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-1-11e651f1a058> in <module>()
      1 from os import listdir
      2 from os.path import isfile, join
----> 3 from sklearn.metrics import normalized_mutual_info_score

ImportError: No module named 'sklearn'

In [11]:

    

test_file = "karate_metadata.txt"
with open(test_file) as f:
    test_set = []
    for line in f.readlines():
        a, b = line.split()
        b = int(b)
        test_set.append(b)
print test_set


    

    




[1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 1, 1, 2, 2, 1, 1, 2, 1, 2, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2]

In [20]:

    

onlyfiles = [ f for f in listdir('./') if 'comm' in f]
train_set = [0] * len(test_set)
print(onlyfiles)
for file in onlyfiles:
    
    with open(file) as f:
        now_label = 0
        for line in f.readlines():
            now_label += 1
            node_list = line.split()
            for node in node_list:
                node = int(node)
                train_set[node - 1] = now_label
    print(normalized_mutual_info_score(test_set, train_set))
    print(train_set)


    

    




['comm3', 'comm4', 'comm5']
0.22776171119
[3, 1, 1, 1, 3, 3, 3, 1, 1, 0, 3, 0, 1, 1, 1, 1, 3, 1, 1, 1, 1, 1, 0, 1, 2, 2, 1, 1, 1, 1, 1, 2, 1, 1]
0.33707662446
[3, 3, 3, 3, 3, 3, 3, 3, 2, 0, 3, 0, 1, 3, 1, 1, 3, 1, 1, 1, 1, 1, 0, 1, 2, 2, 1, 1, 1, 1, 2, 2, 2, 2]
0.186169626583
[1, 1, 1, 1, 3, 3, 3, 1, 2, 0, 3, 0, 1, 1, 1, 1, 3, 1, 1, 1, 1, 1, 0, 1, 2, 2, 1, 1, 1, 1, 2, 2, 2, 2]

In [17]:

    




    

    




0.186169626583