我有一組評論,我用 k-means 對它們進行了聚類,并獲得了每個評論所屬的聚類(例如:1,2,3...)。我還有這些屬于哪些集群的真實標簽,例如:位置、食物等),我需要將它們與蘭德指數進行比較。
由于我有簇號和簇標簽,我如何應用 Rand 索引進行比較?
我應該遵循任何中間步驟嗎?
編輯:我看過后蘭德指數函式(聚類性能評估),但它沒有回答我的問題。
在那個問題中,你有
labels_true = [1, 1, 0, 0, 0, 0]
labels_pred = [0, 0, 0, 1, 0, 1]
但我所擁有的是下面的東西,
labels_true = ['food', 'view', 'room', 'food', 'staff', 'staff']
labels_pred = [0, 0, 0, 1, 0, 1]
任何幫助都受到高度贊賞。
uj5u.com熱心網友回復:
只需使用該sklearn.metrics.rand_score功能:
from sklearn.metrics import rand_score
rand_score(labels_true, labels_pred)
真實標簽和預測標簽是否在不同域中具有值并不重要。請看一下示例:
>>> rand_score(['a', 'b', 'c'], [5, 6, 7])
1.0
>>> rand_score([0, 1, 2], [5, 6, 7])
1.0
>>> rand_score(['a', 'a', 'b'], [0, 1, 2])
0.6666666666666666
>>> rand_score(['a', 'a', 'b'], [7, 7, 2])
1.0
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/366014.html
