diff --git a/learning.py b/learning.py index df5d6fce3..bf0576e74 100644 --- a/learning.py +++ b/learning.py @@ -365,7 +365,7 @@ def plurality_value(examples): def count(attr, val, examples): """Count the number of examples that have attr = val.""" - return len(e[attr] == val for e in examples) #count(e[attr] == val for e in examples) + return sum(e[attr] == val for e in examples) #count(e[attr] == val for e in examples) def all_same_class(examples): """Are all these examples in the same target class?"""