hexviz / tests /test_attention.py

Commit History

Split attention.py into attention.py and models.py
eb9ae1f

aksell commited on

Add ZymCTRL model
cfba77f

aksell commited on

Rename project protention -> hexviz
28d4b1e

aksell commited on

Visualize attention pairs on structure
58c7b8d

aksell commited on

Add ModelType enum and Model class to hold layers and head count
ebbe380

aksell commited on

Implement get_attention for tape BERT
466a8f2

aksell commited on

Add get_sequence
b7ab123

aksell commited on

Add get_protT5
87c0dbc

aksell commited on

Add get_structure
f8402f9

aksell commited on