#ANSQ1
data = np.load("data-2class.npz")
d = data['d']
l = data['l']

def sigma(a):
    return 1./(1.+np.exp(-a))

w = [0,-1,-1]

probability = 1

for n in range(len(l)):
    if int(l[n]) == 0:
        update = sigma(w[0]+w[1]*d[n,0]+w[2]*d[n,1])

    if int(l[n]) == 1:
        update = 1 - sigma(w[0]+w[1]*d[n,0]+w[2]*d[n,1])

    probability = update * probability

print('The total probability is equal to:')
print(probability)
#/ANSQ1