@@ -275,7 +275,7 @@ def confusion_matrix(y_true, y_pred, labels=None, sample_weight=None):
275275 return CM
276276
277277
278- def cohen_kappa_score (y1 , y2 , labels = None , weights = None ):
278+ def cohen_kappa_score (y1 , y2 , labels = None , weights = None , sample_weight = None ):
279279 """Cohen's kappa: a statistic that measures inter-annotator agreement.
280280
281281 This function computes Cohen's kappa [1]_, a score that expresses the level
@@ -311,6 +311,9 @@ class labels [2]_.
311311 List of weighting type to calculate the score. None means no weighted;
312312 "linear" means linear weighted; "quadratic" means quadratic weighted.
313313
314+ sample_weight : array-like of shape = [n_samples], optional
315+ Sample weights.
316+
314317 Returns
315318 -------
316319 kappa : float
@@ -328,7 +331,8 @@ class labels [2]_.
328331 .. [3] `Wikipedia entry for the Cohen's kappa.
329332 <https://en.wikipedia.org/wiki/Cohen%27s_kappa>`_
330333 """
331- confusion = confusion_matrix (y1 , y2 , labels = labels )
334+ confusion = confusion_matrix (y1 , y2 , labels = labels ,
335+ sample_weight = sample_weight )
332336 n_classes = confusion .shape [0 ]
333337 sum0 = np .sum (confusion , axis = 0 )
334338 sum1 = np .sum (confusion , axis = 1 )
0 commit comments