def test_mutual_information(self): labels_true, labels_pred = _linearize(self.labels_true, self.labels_pred) mi = self.metrics._mutual_information(self.metrics._clusters_pred, self.metrics._clusters_true) self.assertAlmostEqual( mi, skmetrics.mutual_info_score(labels_true, labels_pred))
def test_homogeneity_completeness_vmeasure(self): labels_true, labels_pred = _linearize(self.labels_true, self.labels_pred) sk_homogeneity, sk_completeness, sk_vmeasure = skmetrics.homogeneity_completeness_v_measure(labels_true, labels_pred) homogeneity, completeness, vmeasure = self.metrics._homogeneity_completeness_vmeasure(1) self.assertEqual(homogeneity, sk_homogeneity) self.assertEqual(completeness, sk_completeness) self.assertEqual(sk_vmeasure, vmeasure)
def test_homogeneity_completeness_vmeasure(self): labels_true, labels_pred = _linearize(self.labels_true, self.labels_pred) sk_homogeneity, sk_completeness, sk_vmeasure = skmetrics.homogeneity_completeness_v_measure( labels_true, labels_pred) homogeneity, completeness, vmeasure = self.metrics._homogeneity_completeness_vmeasure( 1) self.assertEqual(homogeneity, sk_homogeneity) self.assertEqual(completeness, sk_completeness) self.assertEqual(sk_vmeasure, vmeasure)
def test_mutual_information(self): labels_true, labels_pred = _linearize(self.labels_true, self.labels_pred) mi = self.metrics._mutual_information(self.metrics._clusters_pred, self.metrics._clusters_true) self.assertAlmostEqual(mi, skmetrics.mutual_info_score(labels_true, labels_pred))