├── .gitignore ├── LICENSE ├── README.md ├── assets ├── comparison.png ├── getting_started_output.png └── logo.png ├── at2 ├── __init__.py ├── attribution │ ├── __init__.py │ ├── attention.py │ ├── attributors.py │ ├── features.py │ ├── score_estimators.py │ ├── trainer.py │ └── utils.py ├── tasks │ ├── __init__.py │ ├── base.py │ ├── chat.py │ ├── context_attribution.py │ └── thought_attribution.py └── utils.py ├── notebooks ├── getting_started.ipynb ├── multi_document_context_attribution.ipynb ├── thought_attribution.ipynb └── train_at2.ipynb ├── pyproject.toml ├── scripts ├── train_context_at2.py ├── train_thought_at2.py └── utils.py └── tests ├── __init__.py └── attribution ├── __init__.py ├── test_attention.py ├── test_attributors.py └── test_trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/README.md -------------------------------------------------------------------------------- /assets/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/assets/comparison.png -------------------------------------------------------------------------------- /assets/getting_started_output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/assets/getting_started_output.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/assets/logo.png -------------------------------------------------------------------------------- /at2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/__init__.py -------------------------------------------------------------------------------- /at2/attribution/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/__init__.py -------------------------------------------------------------------------------- /at2/attribution/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/attention.py -------------------------------------------------------------------------------- /at2/attribution/attributors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/attributors.py -------------------------------------------------------------------------------- /at2/attribution/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/features.py -------------------------------------------------------------------------------- /at2/attribution/score_estimators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/score_estimators.py -------------------------------------------------------------------------------- /at2/attribution/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/trainer.py -------------------------------------------------------------------------------- /at2/attribution/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/attribution/utils.py -------------------------------------------------------------------------------- /at2/tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/tasks/__init__.py -------------------------------------------------------------------------------- /at2/tasks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/tasks/base.py -------------------------------------------------------------------------------- /at2/tasks/chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/tasks/chat.py -------------------------------------------------------------------------------- /at2/tasks/context_attribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/tasks/context_attribution.py -------------------------------------------------------------------------------- /at2/tasks/thought_attribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/tasks/thought_attribution.py -------------------------------------------------------------------------------- /at2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/at2/utils.py -------------------------------------------------------------------------------- /notebooks/getting_started.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/notebooks/getting_started.ipynb -------------------------------------------------------------------------------- /notebooks/multi_document_context_attribution.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/notebooks/multi_document_context_attribution.ipynb -------------------------------------------------------------------------------- /notebooks/thought_attribution.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/notebooks/thought_attribution.ipynb -------------------------------------------------------------------------------- /notebooks/train_at2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/notebooks/train_at2.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/train_context_at2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/scripts/train_context_at2.py -------------------------------------------------------------------------------- /scripts/train_thought_at2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/scripts/train_thought_at2.py -------------------------------------------------------------------------------- /scripts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/scripts/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/attribution/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/attribution/test_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/tests/attribution/test_attention.py -------------------------------------------------------------------------------- /tests/attribution/test_attributors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/tests/attribution/test_attributors.py -------------------------------------------------------------------------------- /tests/attribution/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MadryLab/AT2/HEAD/tests/attribution/test_trainer.py --------------------------------------------------------------------------------