Neuronová sít v PyBrain

Příklad: Vytvoření jednoduchého XOR problému řešeného pomocí neuronové sítě v PyBrain

Představme si, že chceme naučit neuronovou síť řešit XOR problém. XOR je logická operace, která má následující pravdivostní tabulku:

A | B | Výstup

0 | 0 | 0

0 | 1 | 1

1 | 0 | 1

1 | 1 | 0

Postup:

Importování knihoven:

Vytvoření neuronové sítě:

Vytvoření datasetu pro trénink:

Trénování neuronové sítě pomocí algoritmu zpětného šíření chyby (backpropagation):

Testování naučené neuronové sítě na vstupních datech:

V tomto příkladě jsme vytvořili jednoduchou neuronovou síť s jednou skrytou vrstvou, která řeší XOR problém. Neuronová síť má 2 vstupní neurony, 2 neurony ve skryté vrstvě a 1 výstupní neuron. Pro trénink neuronové sítě jsme použili algoritmus zpětného šíření chyby (backpropagation) a tréninkový dataset obsahuje čtyři vzory pro všechny možné kombinace vstupních hodnot A a B. Poté, co jsme neuronovou síť natrénovali, jsme ji otestovali na vstupních datech a zkontrolovali, zda správně předpovídá výstupy pro XOR problém.

[bsa_pro_ad_space id=7]