├── README.md ├── TVeval.py ├── TVframework.py ├── agg_data ├── ag_news │ └── 0.json ├── antonym │ └── 0.json ├── english-french │ └── 0.json ├── person-instrument │ └── 0.json ├── person-occupation │ └── 0.json └── product-company │ └── 0.json ├── aggregation.py ├── data ├── ag_news │ ├── 0.json │ ├── 1.json │ └── 2.json ├── antonym │ ├── 0.json │ ├── 1.json │ └── 2.json ├── capitalize │ ├── 0.json │ ├── 1.json │ └── 2.json ├── country-capital │ ├── 0.json │ ├── 1.json │ └── 2.json ├── english-french │ ├── 0.json │ ├── 1.json │ └── 2.json ├── landmark-country │ ├── 0.json │ ├── 1.json │ └── 2.json ├── person-instrument │ ├── 0.json │ ├── 1.json │ └── 2.json ├── person-occupation │ ├── 0.json │ ├── 1.json │ └── 2.json ├── person-sport │ ├── 0.json │ ├── 1.json │ └── 2.json ├── present-past │ ├── 0.json │ ├── 1.json │ └── 2.json ├── product-company │ ├── 0.json │ ├── 1.json │ └── 2.json └── singular-plural │ ├── 0.json │ ├── 1.json │ └── 2.json ├── function_vector_implement.py ├── optimizer.py ├── requirements.txt ├── script ├── gptj_aggregation.sh ├── gptj_function_vector.sh ├── gptj_optimization.sh ├── llama2_aggregation.sh ├── llama2_function_vector.sh └── llama2_optimization.sh ├── selected_layers.py ├── state_vector_optimize.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/README.md -------------------------------------------------------------------------------- /TVeval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/TVeval.py -------------------------------------------------------------------------------- /TVframework.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/TVframework.py -------------------------------------------------------------------------------- /agg_data/ag_news/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/agg_data/ag_news/0.json -------------------------------------------------------------------------------- /agg_data/antonym/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/agg_data/antonym/0.json -------------------------------------------------------------------------------- /agg_data/english-french/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/agg_data/english-french/0.json -------------------------------------------------------------------------------- /agg_data/person-instrument/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/agg_data/person-instrument/0.json -------------------------------------------------------------------------------- /agg_data/person-occupation/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/agg_data/person-occupation/0.json -------------------------------------------------------------------------------- /agg_data/product-company/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/agg_data/product-company/0.json -------------------------------------------------------------------------------- /aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/aggregation.py -------------------------------------------------------------------------------- /data/ag_news/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/ag_news/0.json -------------------------------------------------------------------------------- /data/ag_news/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/ag_news/1.json -------------------------------------------------------------------------------- /data/ag_news/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/ag_news/2.json -------------------------------------------------------------------------------- /data/antonym/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/antonym/0.json -------------------------------------------------------------------------------- /data/antonym/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/antonym/1.json -------------------------------------------------------------------------------- /data/antonym/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/antonym/2.json -------------------------------------------------------------------------------- /data/capitalize/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/capitalize/0.json -------------------------------------------------------------------------------- /data/capitalize/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/capitalize/1.json -------------------------------------------------------------------------------- /data/capitalize/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/capitalize/2.json -------------------------------------------------------------------------------- /data/country-capital/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/country-capital/0.json -------------------------------------------------------------------------------- /data/country-capital/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/country-capital/1.json -------------------------------------------------------------------------------- /data/country-capital/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/country-capital/2.json -------------------------------------------------------------------------------- /data/english-french/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/english-french/0.json -------------------------------------------------------------------------------- /data/english-french/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/english-french/1.json -------------------------------------------------------------------------------- /data/english-french/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/english-french/2.json -------------------------------------------------------------------------------- /data/landmark-country/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/landmark-country/0.json -------------------------------------------------------------------------------- /data/landmark-country/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/landmark-country/1.json -------------------------------------------------------------------------------- /data/landmark-country/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/landmark-country/2.json -------------------------------------------------------------------------------- /data/person-instrument/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-instrument/0.json -------------------------------------------------------------------------------- /data/person-instrument/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-instrument/1.json -------------------------------------------------------------------------------- /data/person-instrument/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-instrument/2.json -------------------------------------------------------------------------------- /data/person-occupation/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-occupation/0.json -------------------------------------------------------------------------------- /data/person-occupation/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-occupation/1.json -------------------------------------------------------------------------------- /data/person-occupation/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-occupation/2.json -------------------------------------------------------------------------------- /data/person-sport/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-sport/0.json -------------------------------------------------------------------------------- /data/person-sport/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-sport/1.json -------------------------------------------------------------------------------- /data/person-sport/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/person-sport/2.json -------------------------------------------------------------------------------- /data/present-past/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/present-past/0.json -------------------------------------------------------------------------------- /data/present-past/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/present-past/1.json -------------------------------------------------------------------------------- /data/present-past/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/present-past/2.json -------------------------------------------------------------------------------- /data/product-company/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/product-company/0.json -------------------------------------------------------------------------------- /data/product-company/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/product-company/1.json -------------------------------------------------------------------------------- /data/product-company/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/product-company/2.json -------------------------------------------------------------------------------- /data/singular-plural/0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/singular-plural/0.json -------------------------------------------------------------------------------- /data/singular-plural/1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/singular-plural/1.json -------------------------------------------------------------------------------- /data/singular-plural/2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/data/singular-plural/2.json -------------------------------------------------------------------------------- /function_vector_implement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/function_vector_implement.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/optimizer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/gptj_aggregation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/script/gptj_aggregation.sh -------------------------------------------------------------------------------- /script/gptj_function_vector.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/script/gptj_function_vector.sh -------------------------------------------------------------------------------- /script/gptj_optimization.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/script/gptj_optimization.sh -------------------------------------------------------------------------------- /script/llama2_aggregation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/script/llama2_aggregation.sh -------------------------------------------------------------------------------- /script/llama2_function_vector.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/script/llama2_function_vector.sh -------------------------------------------------------------------------------- /script/llama2_optimization.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/script/llama2_optimization.sh -------------------------------------------------------------------------------- /selected_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/selected_layers.py -------------------------------------------------------------------------------- /state_vector_optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/state_vector_optimize.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HITsz-TMG/ICL-State-Vector/HEAD/utils.py --------------------------------------------------------------------------------