From df14e94a680b639c6b400855589aa4b2962911d1 Mon Sep 17 00:00:00 2001 From: Rishabh Date: Wed, 8 Mar 2017 21:43:45 +0530 Subject: [PATCH] Fixed a bug in Decision Tree Learner --- learning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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?"""