jbloomAus / DecisionTransformerInterpretability

Interpreting how transformers simulate agents performing RL tasks
https://jbloomaus-decisiontransformerinterpretability-app-4edcnc.streamlit.app/
MIT License
61 stars 15 forks source link

Added text to cosine embedding heatmap of size 10 or less. Increased font size. #100

Closed JayBaileyCS closed 11 months ago

JayBaileyCS commented 11 months ago

Size 10: image Size 11: image

I set it to white text if value > 0.5 because black on dark blue is hard to see, but white on light blue is even worse. So we use white for dark blue and black for everything else. Not consistent design but being able to see numbers is more important and we can't do that with just black or white, and colored text would be much more inconsistent with other graphs.