Skip to content

Commit 5316898

Browse files
TheSPARTAnorvig
authored andcommitted
Fixed a bug in Decision Tree Learner (aimacode#334)
1 parent bf15924 commit 5316898

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

learning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,7 @@ def plurality_value(examples):
376376

377377
def count(attr, val, examples):
378378
"""Count the number of examples that have attr = val."""
379-
return len(e[attr] == val for e in examples) #count(e[attr] == val for e in examples)
379+
return sum(e[attr] == val for e in examples) #count(e[attr] == val for e in examples)
380380

381381
def all_same_class(examples):
382382
"""Are all these examples in the same target class?"""

0 commit comments

Comments
 (0)