def main(): edges = prep.read_edges(path + fn) net = prep.build_net(edges) train, test = prep.divide_net(net, ratio) #CN sim = CN.predict_link(train) #Jaccard sim = Jaccard.predict_link(train) #AA sim = AA.predict_link(train) #RA sim = RA.predict_link(train) #PA sim = PA.predict_link(train) #Katz sim = Katz.predict_link(train) #SimRank sim = SimRank.predict_link(train)
def test(): edges = prep.read_edges(path + in_fn) net = prep.build_net(edges) my = SimRank.predict_link(net, 0.8) output_mat(my, path + 'my.txt') oth = read_mat(path + 'oth.txt') print cmp_mat(my, oth)