From 20af68fa1bc87ad70a24820e7c28b45fa6a4a452 Mon Sep 17 00:00:00 2001 From: Pring Wong Date: Sun, 6 May 2018 22:24:37 +0800 Subject: [PATCH] Update decision_tree.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 简化set --- decision_tree/decision_tree.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/decision_tree/decision_tree.py b/decision_tree/decision_tree.py index d1f837f..d9b650a 100644 --- a/decision_tree/decision_tree.py +++ b/decision_tree/decision_tree.py @@ -72,7 +72,7 @@ def calc_ent(x): calculate shanno ent of x """ - x_value_list = set([x[i] for i in range(x.shape[0])]) + x_value_list = set(x) ent = 0.0 for x_value in x_value_list: p = float(x[x == x_value].shape[0]) / x.shape[0] @@ -87,7 +87,7 @@ def calc_condition_ent(x, y): """ # calc ent(y|x) - x_value_list = set([x[i] for i in range(x.shape[0])]) + x_value_list = set(x) ent = 0.0 for x_value in x_value_list: sub_y = y[x == x_value]