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))
