single root restriction

README.md

@@ -75,6 +75,8 @@ class TransformersUD(object):
       s,e=d.start_logits.tolist()[0],d.end_logits.tolist()[0]
       for j in range(n):
         m[i+1,0 if i==j else j+1]=s[b[j]]+e[b[j+1]-1]
+    i=numpy.nanargmax(m[:,0])
+    m[0:i,0]=m[i+1:,0]=numpy.nan
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     u="# text = "+text.replace("\n"," ")+"\n"
     for i,(s,e,p) in enumerate(w,1):