algorithm improved

ud.py

@@ -69,10 +69,12 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
           k+=1
         k+=1
     g=self.model.config.label2id["X|x|r-goeswith"]
-    r=numpy.tri(e.shape[0])
+    m,r=numpy.max(e,axis=2),numpy.tri(e.shape[0])
     for i in range(e.shape[0]):
       for j in range(i+2,e.shape[1]):
-        r[i,j]=r[i,j-1] if numpy.argmax(e[i,j-1])==g else 1
+        r[i,j]=1
+        if numpy.argmax(e[i,j-1])==g and numpy.argmax(m[:,j-1])==i:
+          r[i,j]=r[i,j-1]
     e[:,:,g]+=numpy.where(r==0,0,-numpy.inf)
     m,p=numpy.max(e,axis=2),numpy.argmax(e,axis=2)
     h=self.chu_liu_edmonds(m)