I didn't make this, but there is a numpy implementation of this algorithm available through scikit-learn in sklearn/utils/linear_assignment_.py . I'm glad I found it, as it seems to be about 4 times faster than the munkres package from PyPI (on which it is based, and which is the python implementation that comes up the most on google).
Reference:
http://stackoverflow.com/questions/1398822/the-assignment-problem-a-numpy-function
No comments:
Post a Comment