2018.1.23
for k in test_set:
yes_mar = 0
yes_edu = 0
yes_def = 0
yes_hsg = 0
no_mar = 0
no_edu = 0
no_def = 0
no_hsg = 0
for t in train_set:
if t[-1] == 0:
if t[0] == k[0]:
no_mar += 1
if t[1] == k[1]:
no_edu += 1
if t[2] == k[2]:
no_def += 1
if t[3] == k[3]:
no_hsg += 1
else:
if t[0] == k[0]:
yes_mar += 1
if t[1] == k[1]:
yes_edu += 1
if t[2] == k[2]:
yes_def += 1
if t[3] == k[3]:
yes_hsg += 1
p_yes = yes_mar/tmp_yes_count*yes_edu/tmp_yes_count*yes_def/tmp_yes_count*yes_hsg/tmp_yes_count*P_yes
p_no = no_mar/tmp_no_count*no_edu/tmp_no_count*no_def/tmp_no_count*no_hsg/tmp_no_count*P_no
if p_yes > p_no:
if k[-1] == 1:
predict += 1
else:
if k[-1] == 0:
predict += 1