This repository contains all necessary code to reproduce the results of arXiv:2210.02226 by Jernej F. Kamenik and Manuel Szewc.
The R&D and BB1 datasets need to be downloaded and preprocessed.
The SA-CWoLA training is performed with this notebook and results are saved in the training/ folder.
Fig. 1 of the preprint is obtained with this notebook and Figs. 2 and 3 with this notebook.