chu_liu_edmonds

ud.py

@@ -51,7 +51,8 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
           x[i]=b(x,i,j)
       if max(x)<0:
         return h
-    y,m=[i for i,j in enumerate(x) if j==max(x)],numpy.full((matrix.shape[0]+1,matrix.shape[1]+1),numpy.nan)
+    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