NMI(標準化互資訊) python實現__python

介紹

NMI是社區發現(community detection)在有標準ground-truth的情況下的重要衡量指標，基本可以比較客觀地評價出一個社區劃分與標準劃分之間相比的準確度。NMI的範圍是0到1，越高代表劃分得越准。具體的原理和例子可以參考這篇部落格：http://www.cnblogs.com/ziqiao/archive/2011/12/13/2286273.html
因為之前自己要用到，但是發現在網上現成的計算NMI的代碼基本都是matlab的和java的，沒找到用python寫的。所以今天閑來無事，把上面部落格中的matlab實現方法用python實現了一下。將來就可以直接用了。 代碼

其實代碼很簡單，就是參考部落格中直接從matlab翻譯成python，不過有一些小坑而已。

直接給代碼，跟部落格中的沒什麼區別啦，這裡解釋一下代碼中的eps是一個很小的正數，為了防止log函數中參數為0而已。
為了方便大家使用python的話可以直接用，最後附有下載地址。 結果

使用代碼中的用例，可以看到，如果輸出的結果是0.36456，說明結果就是正確的。 下載地址

CDNS下載
沒積分的也可以github下載：
github下載連結
記得給star喔~