Python3.7之后，sklearn 的linear_assignment不支持，用scipy代替

最新推荐文章于 2025-04-13 14:26:25 发布

小恩子

最新推荐文章于 2025-04-13 14:26:25 发布

阅读量2.7k

点赞数 7

分类专栏：问题解决文章标签： pytorch 边缘计算深度学习

本文链接：https://blog.youkuaiyun.com/ze1336365763/article/details/121639388

版权

问题解决专栏收录该内容

17 篇文章

订阅专栏

博客介绍了在Python3.7之后sklearn不再支持的情况，以及如何使用Scipy的linear_sum_assignment作为替代。转换过程中需要注意不同包的输出格式差异，并提供了代码示例进行匹配矩阵的转换操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

sklearn在python3.7之后不再支持了，如果还想继续使用from sklearn.utils.linear_assignment_ import linear_assignment，要么降低python版本，要么用scipy代替，但直接代替会出问题，两个包的输出类型和格式不一样，需要做一个转换则可以正常代替了，修改如下：

import numpy as np
from sklearn.utils.linear_assignment_ import linear_assignment
from scipy.optimize import linear_sum_assignment as linear_sum_assignment

const_matrix=np.array([[15,40,45],[20,60,35],[20,40,25]])
matches=linear_assignment(const_matrix)
matches1=linear_sum_assignment(const_matrix)
print("matches_type=",type(matches))
print("matches1_type=",type(matches1))
print("matches=",matches)
print("matches1=",matches1)
print("matches1_T=",np.array(matches1).T)