Added ability to select by one or more states to Embeddings tab.

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 ability to select by one or more states to Embeddings tab. #89

Closed JayBaileyCS closed 11 months ago

JayBaileyCS commented 1 year ago

The actual formatting is pretty ugly, since Streamlit doesn't like spaces (Neither st.text_area nor st.text_input accepted them for me), and separation by commas doesn't work well when the format "channel, (x, y)" has commas. We could use regex but that raises further problems. So instead I decided on the format:

"channel,(x,y)|channel2,(a,b)|channel3,(c,d)" which I am...not super pleased with, but does work. And at least | means "or", which devs will understand.

Open to better ideas here.

JayBaileyCS commented 11 months ago

Fixed according to suggestions