from percettrone import Percettrone
from stampe_video import disegna_funzione, stampa_risultati_single_layer, disegna_grafico_singolo

x = [(0,0),(0,1),(1,0),(1,1)] # Combinazioni
output = (0,0,0,1) # AND Logico
#output = (0,1,1,1) # OR Logico

p = Percettrone(bias=-0.1, lre=0.1)
corrette = 0 #Fermo le epoche se termina prima

soglia_errore_accettabile = 0.001
soglia_funzione_attivazione = 0.5

discriminanti = []

for i in range(1,10000): #Epoche

    if corrette == 4:
        print(f"Epoche necessarie: {i-1}")
        stampa_risultati_single_layer(p)
        break
    
    print(f"EPOCA {i}")
    corrette = 0;

    for j in range(0,4): #Combinazioni
        y = p.funzione_sigmoide(x[j][0], x[j][1])
        
        if y >= soglia_funzione_attivazione:
            previsione = 1
        else:
            previsione = 0

        if previsione == output[j]:
            corrette += 1

        errore = output[j] - y
        
        print("\n")
        disegna_funzione(p, y, x[j][0], x[j][1], True, errore)

        m = -(p.w1 * x[j][0])/p.w2
        q = -(p.bias/p.w2)
        discriminanti.append([m,q])

        if errore != 0:
            appoggio_w1 = errore * y * (1-y) * p.w1 * x[j][0]
            appoggio_w2 = errore * y * (1-y) * p.w1 * x[j][1]
            appoggio_bias = errore * y * (1-y) * p.bias
            p.correggi_pesi(appoggio_w1, appoggio_w2, appoggio_bias)
        else:
            corrette += 1

disegna_grafico_singolo(discriminanti)