improvement for duplicated roots

README.md

@@ -48,6 +48,7 @@ class UDgoeswith(object):
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     if [0 for i in h if i==0]!=[0]:
       m[:,0]+=numpy.where(m[:,0]<numpy.nanmax(m[:,0]),numpy.nan,0)
+      m[[i for i,j in enumerate(h) if j==0]]+=[0 if i==0 or j==0 else numpy.nan for i,j in enumerate(h)]
       h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     u="# text = "+text+"\n"
     v=[(s,e) for s,e in w["offset_mapping"] if s<e]

ud.py

@@ -22,6 +22,8 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     if [0 for i in h if i==0]!=[0]:
       m[:,0]+=numpy.where(m[:,0]<numpy.nanmax(m[:,0]),numpy.nan,0)
+      m[[i for i,j in enumerate(h) if j==0]]+=[0 if i==0 or j==0 else numpy.nan for i,j in enumerate(h)]
+      m[[i for i,j in enumerate(h) if j==0],:]+=[0 if i==0 or j==0 else numpy.nan for i,j in enumerate(h)]
       h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     v=[(s,e) for s,e in model_output["offset_mapping"][0].tolist() if s<e]
     q=[self.model.config.id2label[p[i,j]].split("|") for i,j in enumerate(h)]