-
-
Notifications
You must be signed in to change notification settings - Fork 26.2k
Closed
Labels
Buggood first issueEasy with clear instructions to resolveEasy with clear instructions to resolvehelp wanted
Milestone
Description
from sklearn.datasets import load_iris
X, y = load_iris(return_X_y=True)
from sklearn.tree import DecisionTreeClassifier
tree = DecisionTreeClassifier(criterion='gini')
tree.fit(X, y)
from sklearn.tree import plot_tree
plot_tree(tree)
plots entropy in nodes. It's because entropy
is hard coded here apparently:
https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tree/export.py#L553
Metadata
Metadata
Assignees
Labels
Buggood first issueEasy with clear instructions to resolveEasy with clear instructions to resolvehelp wanted