bug fix

ud.py

@@ -54,8 +54,8 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
     y,x=[i for i,j in enumerate(x) if j==max(x)],[i for i,j in enumerate(x) if j<max(x)]
     z=matrix-numpy.nanmax(matrix,axis=0)
     m=numpy.block([[z[x,:][:,x],numpy.max(z[x,:][:,y],axis=1).reshape(len(x),1)],[numpy.max(z[y,:][:,x],axis=0),numpy.max(z[y,y])]])
-    k=self.chu_liu_edmonds(m)
-    h=[j if i in y else k[x.index(i)] if k[x.index(i)]<len(x) else y[numpy.argmax(z[y,i])] for i,j in enumerate(h)]
+    k=[j if i==len(x) else x[j] if j<len(x) else y[numpy.argmax(z[y,x[i]])] for i,j in enumerate(self.chu_liu_edmonds(m))]
+    h=[j if i in y else k[x.index(i)] for i,j in enumerate(h)]
     i=y[numpy.argmax(z[k[-1],y] if k[-1]<len(x) else z[y,y])]
-    h[i]=k[-1] if k[-1]<len(x) else i
+    h[i]=x[k[-1]] if k[-1]<len(x) else i
     return h