apcamargo / genomad

geNomad: Identification of mobile genetic elements
https://portal.nersc.gov/genomad/
Other
169 stars 17 forks source link

Purpose of the `BranchAttention` class in `igloo.py` #70

Closed a-piece-of-teemo closed 5 months ago

a-piece-of-teemo commented 5 months ago

Hello author, in your igloo.py, I noticed this piece of code, and I am curious if you have used it? Because I saw another similar function, Branch_Attention, in the aggregated_classification module, and I am curious about the relationship between these two functions.

image

apcamargo commented 5 months ago

This is the Keras layer that I used to train the score aggregation model. Because the network is pretty small, I extracted the weights and implemented it in Numpy. You can find the weights and the Numpy implementation here.