search

frederikkemarin / BEND

Benchmarking DNA Language Models on Biologically Meaningful Tasks
BSD 3-Clause "New" or "Revised" License
89 stars 13 forks source link

Enable embedders to return logits #61

Closed fteufel closed 2 weeks ago

fteufel commented 3 weeks ago

So far, embedders only returned embeddings. Logits and crossentropy can also be useful.

First enabled for HyenaDNA. New options for HyenaDNAEmbedder:

return_logits: bool=False, return_loss: bool=False

Other models WIP.

fteufel commented 2 weeks ago