V tomto repozitáry sa nachádza kód neurónovej siete použitej v bakalárskej práci.
Na spustenie projektu je potrebné mať nainštalovaný python s nasledujúcimi kni6nicami:
- pandas
- numpy
- matplotlib.pyplot
- wandb
- torch
- plotly
- sklearn
- tqdm
Pre zaznamenávanie výsledkov pomocou wandb, je potrebné si vytvoriť účet na https://wandb.ai, následne sa pomocou príkazu "wandb login" a použitím vygenerovaného kľúča prihlásiť. Na záver je potrebné vyplniť údaje v kóde pre wandb na vybraných miestach.
Grafy s výsledkami pre jednotlivé stratové funkcie sa nachádzajú v priečinku graphs.