2018.1.23

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:  # no
            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:  # yes
            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
    # print(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
    # print(p_no)
    if p_yes > p_no:
        if k[-1] == 1:
            predict += 1
    else:
        if k[-1] == 0:
            predict += 1