sklearn.metrics.cluster.contingency_matrix?

sklearn.metrics.cluster.contingency_matrix(labels_true, labels_pred, *, eps=None, sparse=False)

源碼

建立描述標簽之間關系的列聯矩陣。

參數 說明
labels_true int array, shape = [n_samples]
真實類標簽可用作參考
labels_pred array-like of shape (n_samples,)
聚類標簽以進行評估
eps None or float, optional.
如果是浮點型，則將該值添加到列聯矩陣中的所有值。這有助于阻止NaN傳播。如果為None，則不進行任何調整。
sparse boolean, optional.
如果為True，則返回一個稀疏的CSR連續性矩陣。如果eps不為None，且稀疏為True，則將引發ValueError。

版本0.18中的新功能。
返回值 說明
contingency {array-like, sparse}, shape=[n_classes_true, n_classes_pred]
矩陣，使得是真實類別i和預測類別j中的樣本數。如果eps為None，則此數組的dtype將為整數。如果給出了eps，則dtype將為float。如果sparse = True，將是scipy.sparse.csr_matrix。