bug fix

ud.py

@@ -51,14 +51,11 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
           x[i]=b(x,i,j)
       if max(x)<0:
         return h
-    y=[i for i,j in enumerate(x) if j==max(x)]
-    m=numpy.full((matrix.shape[0]+1,matrix.shape[1]+1),numpy.nan)
-    m[0:-1,0:-1]=z=matrix-numpy.nanmax(matrix,axis=0)
-    m[0:-1,-1],m[-1,0:-1],m[-1,-1]=numpy.nanmax(z[:,y],axis=1),numpy.nanmax(z[y,:],axis=0),numpy.nanmax(z[y,y])
-    m[y,:]=m[:,y]=numpy.nan
-    m[y,y]=0
+    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)
-    j=y[numpy.nanargmax(z[k[-1],y] if k[-1]<z.shape[0] else z[y,y])]
-    i=k[-1] if k[-1]<z.shape[0] else j
-    z[0:i,j]=z[i+1:,j]=numpy.nan
-    return self.chu_liu_edmonds(z)
+    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)]
+    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
+    return h