├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── images ├── debiasing_the_label_of_two_tower_models.png ├── impl_two_tower_base_retrieval.png ├── two_tower_schematic.png └── user_history_encoder.png ├── setup.py ├── src ├── __init__.py ├── baseline_mips_module.py ├── conftest.py ├── two_tower_base_plus_main_ranker_reward_model.py ├── two_tower_base_retrieval.py ├── two_tower_plus_light_ranker.py ├── two_tower_plus_light_ranker_plus_main_ranker_kd.py ├── two_tower_with_debiasing.py ├── two_tower_with_position_debiased_weights.py ├── two_tower_with_user_debiased_weights.py ├── two_tower_with_user_history_encoder.py └── user_history_encoder.py ├── tests ├── __init__.py ├── conftest.py ├── test_baseline_mips_module.py ├── test_two_tower_base_retrieval.py ├── test_two_tower_user_hist.py ├── test_two_tower_user_hist_position_debias.py └── test_user_history_enc.py └── train ├── __init__.py ├── conftest.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /images/debiasing_the_label_of_two_tower_models.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/images/debiasing_the_label_of_two_tower_models.png -------------------------------------------------------------------------------- /images/impl_two_tower_base_retrieval.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/images/impl_two_tower_base_retrieval.png -------------------------------------------------------------------------------- /images/two_tower_schematic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/images/two_tower_schematic.png -------------------------------------------------------------------------------- /images/user_history_encoder.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/images/user_history_encoder.png -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/baseline_mips_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/baseline_mips_module.py -------------------------------------------------------------------------------- /src/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/conftest.py -------------------------------------------------------------------------------- /src/two_tower_base_plus_main_ranker_reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_base_plus_main_ranker_reward_model.py -------------------------------------------------------------------------------- /src/two_tower_base_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_base_retrieval.py -------------------------------------------------------------------------------- /src/two_tower_plus_light_ranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_plus_light_ranker.py -------------------------------------------------------------------------------- /src/two_tower_plus_light_ranker_plus_main_ranker_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_plus_light_ranker_plus_main_ranker_kd.py -------------------------------------------------------------------------------- /src/two_tower_with_debiasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_with_debiasing.py -------------------------------------------------------------------------------- /src/two_tower_with_position_debiased_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_with_position_debiased_weights.py -------------------------------------------------------------------------------- /src/two_tower_with_user_debiased_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_with_user_debiased_weights.py -------------------------------------------------------------------------------- /src/two_tower_with_user_history_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/two_tower_with_user_history_encoder.py -------------------------------------------------------------------------------- /src/user_history_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/src/user_history_encoder.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_baseline_mips_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/tests/test_baseline_mips_module.py -------------------------------------------------------------------------------- /tests/test_two_tower_base_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/tests/test_two_tower_base_retrieval.py -------------------------------------------------------------------------------- /tests/test_two_tower_user_hist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/tests/test_two_tower_user_hist.py -------------------------------------------------------------------------------- /tests/test_two_tower_user_hist_position_debias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/tests/test_two_tower_user_hist_position_debias.py -------------------------------------------------------------------------------- /tests/test_user_history_enc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/tests/test_user_history_enc.py -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/train/conftest.py -------------------------------------------------------------------------------- /train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gauravchak/two_tower_models/HEAD/train/train.py --------------------------------------------------------------------------------