Fix entroy and ig

This commit is contained in:
2022-11-28 18:03:46 +01:00
parent e9e2a66203
commit 6c507a24b0
7 changed files with 34 additions and 15 deletions

View File

@@ -7,7 +7,9 @@ X = data.data
y = data.target
features = data.feature_names
test = FImdlp()
# Xcutpoints = test.fit(X, y, features=features).transform(X)
Xcutpoints = test.fit(X, y, features=features).transform(X)
clf = CFImdlp(debug=True)
print("Cut points for feature 0 in Iris dataset:")
print(clf.cut_points(X[:, 0], y))
print("Xcut")
print(Xcutpoints)