NetName = 'PB'
elif Set == 5:
NetFile = u'Data/INT.txt'
NetName = 'INT'
elif Set == 6:
NetFile = u'Data/USAir.txt'
NetName = 'USAir'
else:
print 'Input Error'
print "\nLink Prediction start：\n"
TrainFile_Path = 'Data\\'+NetName+'\\Train.txt'
if os.path.exists(TrainFile_Path):
Train_File = 'Data\\'+NetName+'\\Train.txt'
Test_File = 'Data\\'+NetName+'\\Test.txt'
MatrixAdjacency_Train,MatrixAdjacency_Test,MaxNodeNum = Initialize.Init2(Test_File, Train_File)
else:
MatrixAdjacency_Net,MaxNodeNum = Initialize.Init(NetFile)
MatrixAdjacency_Train,MatrixAdjacency_Test = Initialize.Divide(NetFile, MatrixAdjacency_Net, MaxNodeNum,NetName)
#计算相似性矩阵
# READEME = '''\nPlease choose a method:
# CN 0
# Salton 1
# Jaccard 2
# Sorenson 3
# HPI 4
# HDI 5
# LHN-I 6
# PA 7
# AA 8
