Repository for the model training.
Original Repo with ongoing development: here
To reproduce publication model:
- Make sure your environment reflects requirement.txt
- Run
generate_data.pyand input the number of traces (we used 250k initially, which ends up being ~150k after balancing) - Run
train_model.py. Be sure to setexclude_alex_fret=Falsefor the full model.