Doctorado-ML/mufs
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/mufs.git synced 2025-08-18 00:55:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix some tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-06-01 12:50:35 +02:00
parent 794374fe8c
commit b15a059b1d
3 changed files with 31 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
mfs/tests/Metrics_test.py
View File

@@ -62,7 +62,7 @@ class Metrics_test(unittest.TestCase):
7.6373991502818175,
]
n_samples, n_features = self.X_w_c.shape
for c, res_expected in zip(range(n_features), expected):
for c, res_expected in enumerate(expected):
computed = metric.differential_entropy(
self.X_w_c[:, c], n_samples - 1
)
@@ -76,8 +76,10 @@ class Metrics_test(unittest.TestCase):
0.15663362014829718,
0.13032469395094992,
]
for expected, col in zip(results_expected, range(self.n)):
computed = metric.conditional_entropy(self.X_i[:, col], self.y, 3)
for expected, col in zip(results_expected, range(self.X_i.shape[1])):
computed = metric.conditional_entropy(
self.X_i[:, col], self.y_i, 3
)
self.assertAlmostEqual(expected, computed)
self.assertAlmostEqual(
0.6309297535714573,
@@ -99,8 +101,8 @@ class Metrics_test(unittest.TestCase):
0.8433663798517026,
0.8696753060490499,
]
for expected, col in zip(results_expected, range(self.n)):
computed = metric.information_gain(self.X_i[:, col], self.y, 3)
for expected, col in zip(results_expected, range(self.X_i.shape[1])):
computed = metric.information_gain(self.X_i[:, col], self.y_i, 3)
self.assertAlmostEqual(expected, computed)
# https://planetcalc.com/8419/
# ?_d=FrDfFN2COAhqh9Pb5ycqy5CeKgIOxlfSjKgyyIR.Q5L0np-g-hw6yv8M1Q8_
@@ -110,8 +112,8 @@ class Metrics_test(unittest.TestCase):
1.336704086,
1.378402748,
]
for expected, col in zip(results_expected, range(self.n)):
computed = metric.information_gain(self.X_i[:, col], self.y, 2)
for expected, col in zip(results_expected, range(self.X_i.shape[1])):
computed = metric.information_gain(self.X_i[:, col], self.y_i, 2)
self.assertAlmostEqual(expected, computed)
def test_symmetrical_uncertainty(self):
@@ -122,21 +124,23 @@ class Metrics_test(unittest.TestCase):
0.810724587460511,
0.870521418179061,
]
for expected, col in zip(results_expected, range(self.n)):
computed = metric.symmetrical_uncertainty(self.X_i[:, col], self.y)
for expected, col in zip(results_expected, range(self.X_i.shape[1])):
computed = metric.symmetrical_uncertainty(
self.X_i[:, col], self.y_i
)
self.assertAlmostEqual(expected, computed)
def test_symmetrical_uncertainty_continuous(self):
metric = Metrics()
results_expected = [
0.33296547388990266,
0.19068147573570668,
0.810724587460511,
0.870521418179061,
-0.08368315199022527,
-0.08539330663499867,
-0.026524185532893957,
-0.016238166071083728,
]
for expected, col in zip(results_expected, range(self.n)):
for expected, col in zip(results_expected, range(self.X_w.shape[1])):
computed = metric.symmetrical_unc_continuous(
self.X_i[:, col], self.y
self.X_w_c[:, col], self.y_w
)
print(computed)
# self.assertAlmostEqual(expected, computed)
# print(computed)
self.assertAlmostEqual(expected, computed)