add entropy

2020-05-19 19:43:39 +02:00 · 2020-05-19 19:43:39 +02:00 · f49d0069c1
commit f49d0069c1
parent a68cdb94c0
1 changed files with 1 additions and 1 deletions
--- a/Tree/decision_tree.py
+++ b/Tree/decision_tree.py
@ -2,6 +2,6 @@ from sklearn import tree


 def make_tree(maps, actions):
-    clf = tree.DecisionTreeClassifier()
+    clf = tree.DecisionTreeClassifier(criterion="entropy")
    clf = clf.fit(maps, actions)
    return clf