first try with sklearn mi

This commit is contained in:
2021-06-02 01:46:21 +02:00
parent eb00e1516a
commit 96098e9fe1
2 changed files with 95 additions and 7 deletions

View File

@@ -17,6 +17,14 @@ print(Metrics.information_gain(X, y))
for i in range(n):
print(i, Metrics.information_gain(X[:, i], y))
print("- Continuous features")
# print(Metrics.information_gain_cont(X, y))
print(Metrics.information_gain_cont(X, y))
for i in range(n):
print(i, Metrics.information_gain_cont(X[:, i], y))
print("CFS Discrete")
print(mfsd.cfs(X, y).get_results())
print("CFS continuous")
print(mfsc.cfs(X, y).get_results())
print("FCBF Discrete")
print(mfsd.fcbf(X, y, 1e-7).get_results())
print("FCBF continuous")
print(mfsc.fcbf(X, y, 1e-7).get_results())