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.py
and 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=False
for the full model.