wilburnlab / collage

Codon Likelihihoods Learned Against Genome Evolution (CoLLAGE): a deep learning framework for identifying naturally selected patterns of codon preference within a species
MIT License
1 stars 2 forks source link

Training refactor #2

Closed alope107 closed 1 year ago

alope107 commented 1 year ago

Please review #1 first - this PR builds off of it.

Move all logic except for arg parsing out of the top level script and into the collage package. Add epoch argument to specify number of epochs to train on. Minor formatting lints.