def test_gini_impurity(self):
datas = get_data()
print(gini_impurity(datas))
(set1, set2, compare_op) = divide_set(datas, 'read_faq', False)
print(gini_impurity(set1))
print(gini_impurity(set2))
print(compare_op)
def test_get_keys(self):
keys = set(['location', 'pages_viewed', 'referrer', 'service_chosen', 'read_faq'])
print(keys)
keys.discard('service_chosen')
print(keys)
datas = get_data()
print(datas[0].get_attr_keys())
def test_divide_set(self):
datas = get_data()
print(datas)
(set1, set2, compare_op) = divide_set(datas, 'read_faq', False)
print(set1)
print(set2)
print(compare_op)
def test_display_tree(self):
datas = get_data()
tree = build_tree(datas)
display_tree(tree)
def test_entropy(self):
datas = get_data()
print(entropy(datas))
(set1, set2, compare_value) = divide_set(datas, 'read_faq', False)
print(entropy(set1))
print(entropy(set2))
def test_target_counts(self):
datas = get_data()
print(get_target_counts(datas))
def test_show_data(self):
datas = get_data()
show_data(datas)
def test_get_data(self):
pprint.pprint(get_data())