metrics.py 文件源码

def corr_matr():
    louv = communities_to_array_from_file("./results/you_lovain.txt")
    slm = communities_to_array_from_file("./results/you_slm.txt")
    walk = prepare_comms("./results/you_walktrap.txt")
    label = prepare_comms("./results/you_label_propagation_reduced.txt")
    info = prepare_comms("./results/you_infomap.txt")
    fast = comms_from_file_fast_greedy("./results/com_youtube_ungraph_zero_start_reduced-fc_a.groups")
    fast = communities_to_array(fast)

    data = {'louv': louv, 'slm': slm, 'walk': walk, 'label': label, 'info': info, 'fast': fast}
    #data = {'louv': louv, 'slm': slm, 'walk': walk}
    columns = data.keys()
    df = pd.DataFrame(columns=columns, index=columns)
    for col_a, col_b in itertools.combinations_with_replacement(columns, 2):
        print(col_a, col_b)
        df[col_a][col_b] = calc(data[col_a], data[col_b])
    print(df)
    return df