├── .flake8 ├── .github └── workflows │ └── main.yml ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── data ├── debug_pr.png ├── debug_roc.png ├── debug_shap.png ├── path_config.json ├── prod_model.pickle ├── prod_model_pr.png ├── prod_model_roc.png ├── prod_model_shap.png └── prod_model_v0_1.pickle ├── pyproject.toml ├── requirements.in ├── s2apler ├── __init__.py ├── consts.py ├── data.py ├── eval.py ├── featurizer.py ├── file_cache.py ├── model.py ├── plotting_utils.py ├── text.py └── timo │ ├── README.txt │ ├── __init__.py │ ├── config.yaml │ ├── integration_test.py │ └── interface.py ├── scripts ├── analyze_final_data.ipynb ├── dump_pairwise_model.py ├── experiment.py └── run_ci_locally.sh ├── setup.py ├── tests ├── __init__.py ├── test_cluster.py ├── test_cluster_incremental.py ├── test_data.py ├── test_dataset │ ├── cluster_seeds.json │ ├── clusters.json │ └── papers.json ├── test_eval.py ├── test_featurizer.py └── test_text.py └── uv.lock /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/README.md -------------------------------------------------------------------------------- /data/debug_pr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/debug_pr.png -------------------------------------------------------------------------------- /data/debug_roc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/debug_roc.png -------------------------------------------------------------------------------- /data/debug_shap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/debug_shap.png -------------------------------------------------------------------------------- /data/path_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/path_config.json -------------------------------------------------------------------------------- /data/prod_model.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/prod_model.pickle -------------------------------------------------------------------------------- /data/prod_model_pr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/prod_model_pr.png -------------------------------------------------------------------------------- /data/prod_model_roc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/prod_model_roc.png -------------------------------------------------------------------------------- /data/prod_model_shap.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/prod_model_shap.png -------------------------------------------------------------------------------- /data/prod_model_v0_1.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/data/prod_model_v0_1.pickle -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/requirements.in -------------------------------------------------------------------------------- /s2apler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/__init__.py -------------------------------------------------------------------------------- /s2apler/consts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/consts.py -------------------------------------------------------------------------------- /s2apler/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/data.py -------------------------------------------------------------------------------- /s2apler/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/eval.py -------------------------------------------------------------------------------- /s2apler/featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/featurizer.py -------------------------------------------------------------------------------- /s2apler/file_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/file_cache.py -------------------------------------------------------------------------------- /s2apler/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/model.py -------------------------------------------------------------------------------- /s2apler/plotting_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/plotting_utils.py -------------------------------------------------------------------------------- /s2apler/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/text.py -------------------------------------------------------------------------------- /s2apler/timo/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/timo/README.txt -------------------------------------------------------------------------------- /s2apler/timo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /s2apler/timo/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/timo/config.yaml -------------------------------------------------------------------------------- /s2apler/timo/integration_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/timo/integration_test.py -------------------------------------------------------------------------------- /s2apler/timo/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/s2apler/timo/interface.py -------------------------------------------------------------------------------- /scripts/analyze_final_data.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/scripts/analyze_final_data.ipynb -------------------------------------------------------------------------------- /scripts/dump_pairwise_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/scripts/dump_pairwise_model.py -------------------------------------------------------------------------------- /scripts/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/scripts/experiment.py -------------------------------------------------------------------------------- /scripts/run_ci_locally.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/scripts/run_ci_locally.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_cluster.py -------------------------------------------------------------------------------- /tests/test_cluster_incremental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_cluster_incremental.py -------------------------------------------------------------------------------- /tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_data.py -------------------------------------------------------------------------------- /tests/test_dataset/cluster_seeds.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_dataset/cluster_seeds.json -------------------------------------------------------------------------------- /tests/test_dataset/clusters.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_dataset/clusters.json -------------------------------------------------------------------------------- /tests/test_dataset/papers.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_dataset/papers.json -------------------------------------------------------------------------------- /tests/test_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_eval.py -------------------------------------------------------------------------------- /tests/test_featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_featurizer.py -------------------------------------------------------------------------------- /tests/test_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/tests/test_text.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/S2APLER/HEAD/uv.lock --------------------------------------------------------------------------------