2021SC@SDUSC
继配置好该课题代码实验环境后,小组分工后,后续多篇文章将对 vectorize.py中的代码部分内容进行分析。
代码Link:https://github.com/rikdz/GraphWriter
通过论文大致分析可知,vectorize.py是编码器的一部分,包括原始图向未标记连接图的重构,保留标签信息等。本篇文章将对vectorize.py中dataset类中如下代码进行详细分析。
def __init__(self, args):
args.path = args.datadir + args.data
self.args = args
'''
if args.loadvocab:
with open(args.datadir+"/"+args.loadvocab,'rb') as f:
print("Loading Vocabs")
self.fields = dill.load(f)
self.INP,self.ENT,self.REL,self.TGT,self.OUTP = [x[1] for x in self.fields]
else:
'''
self.mkVocabs(args)
print("Vocab sizes:")
for x in self.fields:
try:
print(x[0],len(