KoichiYasuoka commited on
Commit
9e7a24c
·
1 Parent(s): fac6290

lemma added

Browse files
Files changed (1) hide show
  1. ud.py +1 -1
ud.py CHANGED
@@ -127,7 +127,7 @@ class UniversalDependenciesPipeline(BellmanFordTokenClassificationPipeline):
127
  t=model_outputs["sentence"].replace("\n"," ")
128
  u="# text = "+t+"\n"
129
  for i,(s,e) in enumerate(off):
130
- u+="\t".join([str(i+1),t[s:e],"_",q[i][0],"_","_" if len(q[i])<3 else "|".join(q[i][1:-1]),str(0 if h[i]==i else h[i]+1),"root" if q[i][-1]=="root" else q[i][-1][2:],"_","_" if i+1<len(off) and e<off[i+1][0] else "SpaceAfter=No"])+"\n"
131
  return u+"\n"
132
  def chu_liu_edmonds(self,matrix):
133
  h=numpy.argmax(matrix,axis=0)
 
127
  t=model_outputs["sentence"].replace("\n"," ")
128
  u="# text = "+t+"\n"
129
  for i,(s,e) in enumerate(off):
130
+ u+="\t".join([str(i+1),t[s:e],t[s:e],q[i][0],"_","_" if len(q[i])<3 else "|".join(q[i][1:-1]),str(0 if h[i]==i else h[i]+1),"root" if q[i][-1]=="root" else q[i][-1][2:],"_","_" if i+1<len(off) and e<off[i+1][0] else "SpaceAfter=No"])+"\n"
131
  return u+"\n"
132
  def chu_liu_edmonds(self,matrix):
133
  h=numpy.argmax(matrix,axis=0)