├── LICENSE ├── README.md ├── configs ├── LLM_Descriptions.json ├── config.yaml └── llm_description_embedding.pkl ├── data ├── GSM8K │ └── GSM8K.json ├── SQUAD │ └── SQUAD.parquet ├── alpaca_data │ └── alpaca_data.json └── multi_news │ └── multi_news.json ├── data_processing ├── construct_router_data.py ├── llm_engine.py ├── multidata_unify.py └── utils.py ├── docs ├── index.html └── static │ ├── css │ ├── bulma-carousel.min.css │ ├── bulma-slider.min.css │ ├── bulma.css.map.txt │ ├── bulma.min.css │ ├── fontawesome.all.min.css │ └── index.css │ ├── images │ ├── Generalization.png │ ├── Main_results.png │ ├── intro_1.png │ └── intro_2.png │ └── js │ ├── bulma-carousel.js │ ├── bulma-carousel.min.js │ ├── bulma-slider.js │ ├── bulma-slider.min.js │ ├── fontawesome.all.min.js │ └── index.js ├── figures └── model.png ├── model ├── graph_nn.py └── multi_task_graph_router.py ├── model_path ├── best_model.pth └── best_model_qa.pth ├── requirements.txt └── run_exp.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/README.md -------------------------------------------------------------------------------- /configs/LLM_Descriptions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/configs/LLM_Descriptions.json -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/llm_description_embedding.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/configs/llm_description_embedding.pkl -------------------------------------------------------------------------------- /data/GSM8K/GSM8K.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data/GSM8K/GSM8K.json -------------------------------------------------------------------------------- /data/SQUAD/SQUAD.parquet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data/SQUAD/SQUAD.parquet -------------------------------------------------------------------------------- /data/alpaca_data/alpaca_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data/alpaca_data/alpaca_data.json -------------------------------------------------------------------------------- /data/multi_news/multi_news.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data/multi_news/multi_news.json -------------------------------------------------------------------------------- /data_processing/construct_router_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data_processing/construct_router_data.py -------------------------------------------------------------------------------- /data_processing/llm_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data_processing/llm_engine.py -------------------------------------------------------------------------------- /data_processing/multidata_unify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data_processing/multidata_unify.py -------------------------------------------------------------------------------- /data_processing/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/data_processing/utils.py -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/static/css/bulma-carousel.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/css/bulma-carousel.min.css -------------------------------------------------------------------------------- /docs/static/css/bulma-slider.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/css/bulma-slider.min.css -------------------------------------------------------------------------------- /docs/static/css/bulma.css.map.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/css/bulma.css.map.txt -------------------------------------------------------------------------------- /docs/static/css/bulma.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/css/bulma.min.css -------------------------------------------------------------------------------- /docs/static/css/fontawesome.all.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/css/fontawesome.all.min.css -------------------------------------------------------------------------------- /docs/static/css/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/css/index.css -------------------------------------------------------------------------------- /docs/static/images/Generalization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/images/Generalization.png -------------------------------------------------------------------------------- /docs/static/images/Main_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/images/Main_results.png -------------------------------------------------------------------------------- /docs/static/images/intro_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/images/intro_1.png -------------------------------------------------------------------------------- /docs/static/images/intro_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/images/intro_2.png -------------------------------------------------------------------------------- /docs/static/js/bulma-carousel.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/js/bulma-carousel.js -------------------------------------------------------------------------------- /docs/static/js/bulma-carousel.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/js/bulma-carousel.min.js -------------------------------------------------------------------------------- /docs/static/js/bulma-slider.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/js/bulma-slider.js -------------------------------------------------------------------------------- /docs/static/js/bulma-slider.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/js/bulma-slider.min.js -------------------------------------------------------------------------------- /docs/static/js/fontawesome.all.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/js/fontawesome.all.min.js -------------------------------------------------------------------------------- /docs/static/js/index.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/docs/static/js/index.js -------------------------------------------------------------------------------- /figures/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/figures/model.png -------------------------------------------------------------------------------- /model/graph_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/model/graph_nn.py -------------------------------------------------------------------------------- /model/multi_task_graph_router.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/model/multi_task_graph_router.py -------------------------------------------------------------------------------- /model_path/best_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/model_path/best_model.pth -------------------------------------------------------------------------------- /model_path/best_model_qa.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/model_path/best_model_qa.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ulab-uiuc/GraphRouter/HEAD/run_exp.py --------------------------------------------------------------------------------